zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 18:11:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
209 Commits 37 Branches 64 Tags 247 MiB
a2bf7693dd
Commit Graph

2 Commits

Author SHA1 Message Date
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
Awni Hannun
dff4a3833f
Module checks the weight on load_weights (#337) 
* update module to check weights on load, also fix docs and reorganize tests

* nits + rebase

* a few more docs updates for Module

* use manual module file

* comment
 2024-01-02 18:55:42 -08:00