mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-26 02:33:21 +08:00
0b28399638
* 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 |
||
|---|---|---|
| .. | ||
| nn | ||
| array.rst | ||
| data_types.rst | ||
| devices_and_streams.rst | ||
| fft.rst | ||
| nn.rst | ||
| ops.rst | ||
| optimizers.rst | ||
| random.rst | ||
| transforms.rst | ||
| tree_utils.rst | ||