jax-dataclasses (pypi): Dataclasses + JAX
Keywords: dataclasses|jax|python