[Issue #1187] Add nan_to_num function initial attempt (#1247)

* initial attempt, working with wrong types * not compiling; mx.float16 and mx.bfloat16 tests added * fix nan to num * nit --------- Co-authored-by: Awni Hannun <[email protected]>
2024-07-25 09:57:37 -07:00
parent baf9fa5f42
commit 5029894662
5 changed files with 93 additions and 1 deletions
@@ -106,6 +106,7 @@ Operations
   minimum
   moveaxis
   multiply
+   nan_to_num
   negative
   not_equal
   ones