grain
Library for reading and processing ML training data.
Installation
To install this package, run one of the following:
Usage Tracking
0.2.15
0.2.14
0.2.12
3 / 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
Nov 26, 2025 at 09:06
License
Apache-2.0
Total Downloads
5.2K
Supported Platforms
linux-64
GitHub Repository
https://github.com/google/grain/Documentation
https://google-grain.readthedocs.io/