Add mx.finfo and use it when making causal mask (#1726)

* finfo * fixes * docs
2025-12-12 15:29:01 +08:00 · 2024-12-19 14:52:41 -08:00
parent e03f0372b1
commit c3628eea49
9 changed files with 154 additions and 3 deletions
--- a/docs/src/python/data_types.rst
+++ b/docs/src/python/data_types.rst
@@ -66,3 +66,4 @@ documentation for more information. Use :func:`issubdtype` to determine if one
   Dtype
   DtypeCategory
   issubdtype
+   finfo