jax-mps

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

Downloads last day: 21
Downloads last week: 311
Downloads last month: 1,239
7-day smoothing