mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-24 17:31:16 +08:00
30bbea2f08
* add gemv masked to JIT plus some fixes * some cleanup * add utils * fix * fix 2 * more cleaning * fix * remove unused mps matmul support * one more nit * revert |
||
|---|---|---|
| .. | ||
| axpby | ||
| mlx_sample_extensions | ||
| bindings.cpp | ||
| CMakeLists.txt | ||
| pyproject.toml | ||
| README.md | ||
| requirements.txt | ||
| setup.py | ||
| test.py | ||
Build
pip install -e .
For faster builds during development, you can also pre-install the requirements:
pip install -r requirements.txt
And then run:
python setup.py build_ext -j8 --inplace
Test
python test.py