mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-24 09:21:16 +08:00
610af352d4
* Dispatch bf16 at run time when using the JIT * fix extension * fix extension build * fix extension build * Update utils.h |
||
|---|---|---|
| .. | ||
| 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