Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
PrototypeCommon

tillahoffmann/jax-mps

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

Owner
tillahoffmann (personal account)
Stars
Stars: 243
Forks
Forks: 32
Language
C++
License
Apache-2.0
Created
2026
Topics
jaxmetal-performance-shadersapple-silicon
Tier
Prototype (100 to 249 ★) What this tier means
View on GitHub

More repos by tillahoffmann

All 2 repos by tillahoffmann

Nearby in C++

All C++ repos