mlx/docs/src/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
..
nn add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) 2024-03-25 12:32:59 -07:00
optimizers Add linear warmup and schedule joining for use with existing schedules (#721) 2024-02-26 07:28:48 -08:00
array.rst add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) 2024-03-25 12:32:59 -07:00
data_types.rst add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) 2024-03-25 12:32:59 -07:00
devices_and_streams.rst Adds device context manager (#679) 2024-02-14 14:14:58 -08:00
fast.rst post nanobind docs fixes and some updates (#889) 2024-03-24 15:03:27 -07:00
fft.rst awni's commit files 2023-11-29 10:30:41 -08:00
linalg.rst QR factorization (#310) 2024-01-26 09:27:31 -08:00
metal.rst bindings for memory info (#761) 2024-03-01 19:51:58 -08:00
nn.rst Common neural network initializers nn.initializers (#456) 2024-01-23 06:47:20 -08:00
ops.rst add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) 2024-03-25 12:32:59 -07:00
optimizers.rst Support LR schedulers (#334) 2024-02-15 11:26:20 -08:00
random.rst Sort some APIs docs by names (a-z) (#472) 2024-01-16 19:37:50 -08:00
transforms.rst Compile docs (#653) 2024-02-08 11:21:50 -08:00
tree_utils.rst jagrit's commit files 2023-11-29 10:52:08 -08:00