Files
mlx/python/tests/test_losses.py
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

11 KiB