Sign in Sign up
pypi

jax-mps

A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.

Activity

Latest release
3d ago
Total releases
204
Cadence
~daily
Last 12 months
204

Reach

Stars
241

Details

License
Apache-2.0
First release
Jan 26, 2026
Releases
Version Released
0.11.2.dev878 pre
0.11.2.dev874 pre
0.11.1 patch
0.11.1.dev872 pre
0.11.1.dev871 pre
0.11.1.dev868 pre
0.11.1.dev866 pre
0.11.1.dev861 pre
0.11.1.dev856 pre
0.11.0 minor
0.11.0.dev854 pre
0.10.12.dev847 pre
0.10.11 patch
0.10.11.dev845 pre
0.10.11.dev843 pre
0.10.11.dev840 pre
0.10.11.dev836 pre
0.10.11.dev834 pre
0.10.11.dev824 pre
0.10.11.dev818 pre
0.10.11.dev815 pre
0.10.10 patch
0.10.10.dev813 pre
0.10.10.dev811 pre
0.10.10.dev809 pre
0.10.10.dev805 pre
0.10.10.dev803 pre
0.10.10.dev800 pre
0.10.10.dev789 pre
0.10.10.dev781 pre
0.10.10.dev779 pre
0.10.10.dev760 pre
0.10.10.dev758 pre
0.10.10.dev756 pre
0.10.9 patch
0.10.9.dev754 pre
0.10.9.dev749 pre
0.10.9.dev748 pre
0.10.9.dev740 pre
0.10.9.dev734 pre
0.10.8 patch
0.10.8.dev732 pre
0.10.8.dev727 pre
0.10.8.dev723 pre
0.10.8.dev719 pre
0.10.7 patch
0.10.7.dev717 pre
0.10.7.dev716 pre
0.10.7.dev713 pre
0.10.7.dev711 pre
1–50 of 204