mlx/python
Asaf Zorea 295ce9db09
Feature expand nn linear (#315)
* Added an identity and bilinear layers
Added a reset_parameters option
Added normal init for bias

* pre-commit run

* add type hints for parameters and the return type
change Bilinear math to x_1 and x_2
change __call__ arguments to x and y instead of input and output
add explanation to the Initialization

* Remove unnecessary reshape

* Added 'i' to bilinear formula

* Changed bilinear computation to two matrix multiplications

* avoid saving intermediate results, kept y in bilinear for better clarity (can be replaced with x1)

* Changed math formula in Linear
Added more explanation to math formulas
Changed x1, x2 reshape to support all inputs sizes
2024-01-02 06:08:53 -08:00
..
mlx Feature expand nn linear (#315) 2024-01-02 06:08:53 -08:00
src Spelling (#342) 2024-01-01 21:08:17 -08:00
tests Feature expand nn linear (#315) 2024-01-02 06:08:53 -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/*