Library for reading and processing ML training data.
conda install conda-forge::grain
Grain is a library for reading data for training and evaluating JAX models. It's open source, fast and deterministic.