mlx/python
Rifur13 2463496471
[Fix] mx.allclose bug with infinite values (#539)
* Added isclose op and fixed comparison with inf values

* Added 'equal_nan' to match numpy

* format

* Add test

* Update python/src/ops.cpp

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* Update python/src/ops.cpp

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* Addressed CR comments

* Update python/src/ops.cpp

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* nits

---------

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>
2024-01-25 20:47:06 -08:00
..
mlx Added Adafactor (#415) 2024-01-23 15:11:27 -08:00
src [Fix] mx.allclose bug with infinite values (#539) 2024-01-25 20:47:06 -08:00
tests [Fix] mx.allclose bug with infinite values (#539) 2024-01-25 20:47:06 -08:00