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.8
Required dependencies: jax | jaxlib

Downloads last day: 179
Downloads last week: 837
Downloads last month: 3,552