zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-21 12:06:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
28009c9cdb
mlx/python
History
__mo_san__ 28009c9cdb Update python/mlx/nn/layers/normalization.py 
Update BatchNorm to support NLC and NHWC input formats

In our convolution operations, we follow the convention that the channels are the last dimension. This commit updates the BatchNorm implementation to support inputs where the channels are the last dimension (NLC or NHWC). This involves changing the dimensions we normalize over and the dimensions we expand our parameters over.

Co-authored-by: Robert McCraith <mccraithrobert@gmail.com>
2023-12-24 23:10:37 +01:00
..
mlx Update python/mlx/nn/layers/normalization.py 2023-12-24 23:10:37 +01:00
src Corrected the example of value_and_grad (#274) 2023-12-23 11:06:38 -08:00
tests refactored and updated batch norm tests ^^ 2023-12-24 23:10:37 +01:00
README.md awni's commit files 2023-11-29 10:30:41 -08:00

README.md

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