Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/google/grain
https://github.com/google/grain
Last synced: about 2 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/google/grain
- Owner: google
- License: apache-2.0
- Created: 2022-08-05T08:29:51.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2024-04-11T22:57:13.000Z (5 months ago)
- Last Synced: 2024-04-12T07:44:54.408Z (5 months ago)
- Language: Python
- Size: 576 KB
- Stars: 181
- Watchers: 9
- Forks: 12
- Open Issues: 62
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
Awesome Lists containing this project
README
# Grain - Feeding JAX Models
Grain is a library for reading data for training and evaluating JAX models. It's
open source, fast and deterministic.* Installation: `pip install grain`
* [Docs](https://github.com/google/grain/tree/main/docs)
* Grain is used by [MaxText](https://github.com/google/maxtext/tree/main), a simple, performant and scalable JAX codebase for LLM.