grain
Library for reading and processing ML training data.
Installation
To install this package, run one of the following:
Usage Tracking
0.2.16
0.2.15
0.2.14
0.2.12
4 / 8 versions selected
Downloads (Last 6 months): 0
Description
Grain is a library for reading data for training and evaluating JAX models. It's open source, fast and deterministic.
About
Summary
Library for reading and processing ML training data.
Last Updated
Mar 2, 2026 at 12:30
License
Apache-2.0
Supported Platforms
linux-64
GitHub Repository
https://github.com/google/grain/Documentation
https://google-grain.readthedocs.io/