tjax

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

Downloads last day: 9
Downloads last week: 231
Downloads last month: 412
7-day smoothing