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