mirror of
https://github.com/ml-explore/mlx.git
synced 2025-09-30 02:38:11 +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
Packaging for PyPI
Install build and twine:
pip install --user --upgrade build
pip install --user --upgrade twine
Generate the source distribution and wheel:
python -m build
Warning use a test server first
Test Upload
Upload to test server:
python -m twine upload --repository testpypi dist/*
Install from test server and check that it works:
python -m pip install --index-url https://test.pypi.org/simple/ --no-deps mlx
Upload
python -m twine upload dist/*