google / grain
Library for reading and processing ML training data.
Why this job: works in the areas we cover, but does not say which job it does
What it does
Grain is a Python library for reading and processing data for training and evaluating JAX models. It is flexible, fast and deterministic.
Get started
No single install command. Read the steps in the README (opens in new tab)
- Maintenance
- Active
- Last commit yesterday
- Stars
- 806
- +5 this week
- Licence
- Apache-2.0
- Allows business use
- Downloads for
- None listed
- No release files
- Latest version
- v0.2.18
- Released 4 months ago
- Built with
- Python
- 96% of the code
Is it looked after?
- Contributors
- 10
- Last 90 days
- Issues
- 69% closed
- 30 still open
- Pull requests
- 642 merged
- 139 waiting
- Releases
- 12
- 4 in the last year
Similar repos
| Repo | Stars | Last commit | Licence | Downloads for |
|---|---|---|---|---|
| magenta / magenta-realtime | 1,866 | 3 months ago | Apache-2.0 | No downloads |
| jax-ml / bonsai | 246 | 5 months ago | Apache-2.0 | No downloads |
| pyro-ppl / funsor | 251 | 3 months ago | Apache-2.0 | No downloads |
More: languages, topics and project record
Languages
- Python 96%
- Starlark 3%
- C++ 1%
- Shell 1%
Topics
- data-pr
- jax
- machine-learning
- python
Project record
- JobOther AI tools
- StatusListed
- Commits964
- Forks98
- Watching11
- Stars in 28 days+14
- CreatedAug 5, 2022
- First seen by usNov 7, 2026
- CheckedNov 12, 2026