mlx/python
Daniel Strobusch 479051ce1c
add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)
* add numeric type hierarchy and issubdtype as well as a set_dtype method to nn.Module with predicate

numeric type hierarchy and issubtype is compatible to the [numpy hierarchy](220f0ab2c5/numpy/_core/numerictypes.py (L42)).

Closes #285.

* nits in docs

* unify type category checking

* nits in docs

* nits in docs

* more docs nits

* fix callable type

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2024-03-25 12:32:59 -07:00
..
mlx add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) 2024-03-25 12:32:59 -07:00
src add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) 2024-03-25 12:32:59 -07:00
tests add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) 2024-03-25 12:32:59 -07:00