tillahoffmann / jax-mpsView on GitHub
A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.
40Feb 17, 2026Updated last week

Alternatives and similar repositories for jax-mps

Users that are interested in jax-mps are comparing it to the libraries listed below

Sorting:

Are these results useful?