Repository navigation

grain

Public

forked from google/grain

About

No description, website, or topics provided.

0 stars

0 watching

Report repository

Name		Name	Last commit message	Last commit date
.github/workflows		.github/workflows
docs		docs
grain		grain
BUILD		BUILD
CONTRIBUTING.md		CONTRIBUTING.md
LICENSE		LICENSE
README.md		README.md
WORKSPACE		WORKSPACE
pyproject.toml		pyproject.toml
setup.py		setup.py

Repository files navigation

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
Grain is used by MaxText, a simple, performant and scalable JAX codebase for LLM.

About

No description, website, or topics provided.

0 stars

0 watching

Report repository

Releases

Packages

Contributors

Languages