tjax

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

Downloads last day: 4
Downloads last week: 79
Downloads last month: 998