tjax

PyPI page
Home page
Author: Neil Girdhar
Summary: Tools for JAX.
Latest version: 1.5.0
Required dependencies: array-api-compat | jax | numpy | optax | optype | rich

Downloads last day: 61
Downloads last week: 253
Downloads last month: 2,334