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
..
mlx_tests.py	copyright + ack	2023-11-30 11:12:53 -08:00
test_array.py	Hashable dtype and mlx.core prefixed repr (#89)	2023-12-09 09:35:28 -08:00
test_autograd.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_bf16.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_blas.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_conv.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_device.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_eval.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_fft.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_load.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_nn.py	added mse_loss, nll_loss and kl_div_loss (#98)	2023-12-09 14:25:03 -08:00
test_ops.py	Add the remainder op (#85)	2023-12-08 15:08:52 -08:00
test_optimizers.py	Pr template (#99)	2023-12-09 09:36:56 -08:00
test_random.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_reduce.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_tree.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_vmap.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00