https://github.com/google/grain

https://github.com/google/grain

Last synced: 26 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 (almost 2 years ago)
• Default Branch: main
• Last Pushed: 2024-04-11T22:57:13.000Z (3 months ago)
• Last Synced: 2024-04-12T07:44:54.408Z (3 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

Lists

• awesome-stars - google/grain - (Python)

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.