Tags → #jax
-
Loadax - High-Performance Data Loading for JAX
A JAX-native dataloading library with prefetching, multi-worker support, and distributed loading for ML training pipelines.
A JAX-native dataloading library with prefetching, multi-worker support, and distributed loading for ML training pipelines.