7 Commits

Author	SHA1	Message	Date
AtomicVar	83f63f2184	Add Margin Ranking Loss (#536)	2024-02-02 10:57:31 -08:00
AtomicVar	755dcf6137	Enable cross_entropy loss to handle dense targets (#517) ... * Enable cross_entropy loss to handle dense targets Dense targets means probabilities or one-hot encodings. * better shape check of weights * nits in docstring --------- Co-authored-by: Awni Hannun <awni@apple.com>	2024-01-23 12:17:22 -08:00
Awni Hannun	7a34e46677	Quantize with groups of 32 (#511) ... * allow quantize with group sizes of 32 * missing cpu dispatch * remove print * Fix qvm for group_size 32 --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>	2024-01-21 06:19:05 -08:00
AtomicVar	550d4bf7c0	Update binary_cross_entropy function to handle both logits and probabilities (#492)	2024-01-18 19:22:23 -08:00
AtomicVar	d1fef34138	Add Gaussian NLL loss function (#477) ... * Add Gaussian NLL loss function --------- Co-authored-by: Awni Hannun <awni@apple.com>	2024-01-18 06:44:44 -08:00
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