mlx/docs/src/python/nn
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
..
functions.rst Fix compile with non standard types (#745) 2024-02-26 19:28:53 -08:00
init.rst Common neural network initializers nn.initializers (#456) 2024-01-23 06:47:20 -08:00
layers.rst Implement RNN, GRU, LSTM (#268) 2024-03-11 21:14:44 -07:00
losses.rst Add Margin Ranking Loss (#536) 2024-02-02 10:57:31 -08:00
module.rst add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) 2024-03-25 12:32:59 -07:00