jax-mps

PyPI page
Home page
Author: None
License: Apache-2.0
Summary: JAX backend for Apple Metal Performance Shaders (MPS)
Latest version: 0.10.0
Required dependencies: jax | jaxlib

Downloads last day: 196
Downloads last week: 532
Downloads last month: 2,425