jax-data (pypi): Native data handling for JAX
Keywords: jax|jax-data|data|dataset|datahandling