5 Commits

Author	SHA1	Message	Date
Enoch Kan	0b28399638	added mse_loss, nll_loss and kl_div_loss (#98) ... * added mse_loss, nll_loss and kl_div_loss * fixed axis not defined error in nll_loss * fixed axis not defined in kl_div_loss * added tests for mse, nll and kl_div * modified docstrings and added reduce helper func * updated docstring in kl_div_loss and moved helper func * added new kl divergence implementation * added reduction to test * updated docstring of kl_div_loss with correct spelling * added losses to nn.rst in docs	2023-12-09 14:25:03 -08:00
Awni Hannun	2520dbcf0a	add losses to the docs, fix black failur (#92)	2023-12-09 06:06:52 -08:00
Jagrit Digani	e6306cfee9	jagrit's commit files	2023-11-29 10:52:08 -08:00
Angelos Katharopoulos	d1f86272a2	angelos's commit files	2023-11-29 10:42:59 -08:00
Awni Hannun	8ca7f9e8e9	awni's commit files	2023-11-29 10:30:41 -08:00