mlx/docs/src/python/nn
YUN, Junwoo 0b8aeddac6
Additoinal losses (#336)
* cosine similarity loss

---------

Co-authored-by: Awni Hannun <awni@apple.com>

* Docstring nits
2024-01-08 14:01:13 -08:00
..
functions.rst Docs updates (#198) 2023-12-17 13:20:55 -08:00
layers.rst implemented InstanceNorm (#244) 2024-01-03 12:21:15 -08:00
losses.rst Additoinal losses (#336) 2024-01-08 14:01:13 -08:00
module.rst Module checks the weight on load_weights (#337) 2024-01-02 18:55:42 -08:00