https://github.com/google/grain

https://github.com/google/grain

Last synced: 14 days ago
JSON representation

• Host: GitHub
• URL: https://github.com/google/grain
• Owner: google
• License: apache-2.0
• Created: 2022-08-05T08:29:51.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2024-08-06T14:55:55.000Z (3 months ago)
• Last Synced: 2024-08-07T10:44:17.797Z (3 months ago)
• Language: Python
• Size: 940 KB
• Stars: 229
• Watchers: 10
• Forks: 17
• Open Issues: 110
• 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.