mlx/python
Angelos Katharopoulos 961435a243
Scatter vjp (#394)
* Add a first scatter vjp
* Implement the scatter_add vjp
* Add array.at to implement user friendly scatters
2024-01-09 13:36:51 -08:00
..
mlx Fix BN stats to not expand shape (#409) 2024-01-09 11:54:51 -08:00
src Scatter vjp (#394) 2024-01-09 13:36:51 -08:00
tests Scatter vjp (#394) 2024-01-09 13:36:51 -08:00
README.md awni's commit files 2023-11-29 10:30:41 -08:00

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/*