zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-15 23:58:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
2770a1024082eb10cce6bc0ac589ad089e7be611
mlx/examples/extensions
History
Awni Hannun c4230747a1 redesign for faster cpu/gpu synch (#1869) 
* redesign for faster cpu/gpu synch

* load + more async CPU

* use command encoder API and move more ops to use it

* make fence back-end generic + CPU only fence

* faster build

* fix async eval

* fixes + handle temporaries

* fix / improve cpu conv

* remove unused status, fix siblings

* fix extensions

* fix

* fix no cpu build

* format

* comments

* fix perf regression, remove unecessary abort

* fix events, task limit cpu

* fix waiting

* fix donation / temporaries in normalization
2025-03-06 19:23:38 -08:00
..
axpby
redesign for faster cpu/gpu synch (#1869)
2025-03-06 19:23:38 -08:00
mlx_sample_extensions
Fix extensions (#1126)
2024-05-16 15:36:25 -07:00
bindings.cpp
Remove "using namespace mlx::core" in benchmarks/examples (#1685)
2024-12-11 07:08:29 -08:00
CMakeLists.txt
redesign for faster cpu/gpu synch (#1869)
2025-03-06 19:23:38 -08:00
pyproject.toml
Bump nanobind to 2.4 + fix (#1710)
2024-12-17 10:57:54 -08:00
README.md
Rename block sparse (#1149)
2024-05-22 07:48:34 -07:00
requirements.txt
Install mlx.dll into the same dir with python bindings on Windows (#1690)
2024-12-13 19:50:39 -08:00
setup.py
bump nanobind + fix extension (#1344)
2024-08-21 16:05:07 -07:00
test.py
Fix extensions (#1126)
2024-05-16 15:36:25 -07:00

README.md

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