mlx/docs/src/python/nn
Awni Hannun fe1dabf272
Fix compile with non standard types (#745)
* refactor tree utils

* fix compile + tree code refactor

* Add an extra test

* add a few missing activations to docs

* hash structure

* Encode the full argument structure

---------

Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
2024-02-26 19:28:53 -08: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 Upsample2d (#414) 2024-02-23 09:55:04 -08:00
losses.rst Add Margin Ranking Loss (#536) 2024-02-02 10:57:31 -08:00
module.rst Compile with capture (#629) 2024-02-07 17:29:22 -08:00