google/grain
Library for reading and processing ML training data.
691 stars and 1,118,889 monthly downloads. Used by 3 other packages. Actively maintained with 18 commits in the last 30 days. Available on PyPI.
Stars
691
Forks
69
Language
Python
License
Apache-2.0
Category
Last pushed
Mar 12, 2026
Monthly downloads
1,118,889
Commits (30d)
18
Dependencies
6
Reverse dependents
3
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/google/grain"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
extropic-ai/thrml
Thermodynamic Hypergraphical Model Library in JAX
patrick-kidger/optimistix
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox....