9 Commits

Author	SHA1	Message	Date
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
Nicholas Santavas	d35fa1db41	Add Hinge, Huber and LogCosh losses (#199)	2023-12-22 10:28:10 -08:00
Awni Hannun	ee0c2835c5	Docs updates (#198) ... Reorganize NN docs + a few other tidbits.	2023-12-17 13:20:55 -08:00
Diogo	02de234ef0	Activations LeakyReLU / PReLU / Softplus / Mish (#109) ... * Leaky_relu / prelu / softplus / mish * added tests * updated bench * remove torch refs, add init to PReLU * added arvix reference to mish * added missing docs	2023-12-11 19:40:57 -08:00
Nicholas Santavas	f5df47ec6e	Add Step, ELU, SELU, Swish activation functions (#117) ... * Add Step, ELU, SELU, Swish activation functions This commit adds the Step, ELU, SELU and Swish activations functions * add to the docs * review	2023-12-11 17:04:07 -08:00
__mo_san__	f4ddd7dc44	Add Binary Cross Entropy loss (#122) ... * update BCE added tests for it ... * added binary cross entropy loss to docs * resolving conflicts for merge	2023-12-11 07:55:18 -08:00
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
Awni Hannun	8ca7f9e8e9	awni's commit files	2023-11-29 10:30:41 -08:00