tillahoffmann

Githubhttp://tillahoffmann.github.io/Followers 111

代码库

tillahoffmann/jax-mps

178

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

C++

apple-siliconjaxmetal-performance-shaders