Package profile
grain
- Summary: Grain: A library for loading and transforming data for ML training.
- Author: Grain team <grain-dev@google.com>
- Homepage: https://github.com/google/grain
- Source: https://github.com/google/grain (Repo profile)
- Number of releases: 13
- First release: 0.1.0 on 2024-04-01
- Latest release: 0.2.11 on 2025-07-02
Releases
PyPI Downloads
Dependencies
Grain has 15 dependencies, 7 of which optional.| Dependency | Optional |
|---|---|
| absl-py | false |
| array-record | false |
| cloudpickle | false |
| dm-tree | false |
| etils | false |
| more-itertools | false |
| numpy | false |
| protobuf | false |
| attrs | true |
| dill | true |
| jax | true |
| jaxlib | true |
| jaxtyping | true |
| pyarrow | true |
| tensorflow-datasets | true |