A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.
VGGT 3D reconstruction optimized for Apple Silicon with sparse attention