jax-dataclasses

PyPI page
Home page
Author: brentyi
License: MIT
Summary: Dataclasses + JAX
Latest version: 1.6.1
Required dependencies: jax | jaxlib | typing-extensions
Optional dependencies: flax | pytest | pytest-cov

Downloads last day: 124
Downloads last week: 2,521
Downloads last month: 13,185