grain
Library for reading and processing ML training data.
Installation
To install this package, run one of the following:
Try This Package Beta
Usage Tracking
0.2.18
0.2.16
0.2.15
0.2.14
0.2.12
5 / 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
Sep 16, 2026 at 21:18
License
Apache-2.0
Supported Platforms
linux-64
linux-aarch64
GitHub Repository
https://github.com/google/grain/Documentation
https://google-grain.readthedocs.io/