Rare find

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

github.com/tillahoffmann/jax-mps

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.