Giters

TalKachman / grainTK

fork of grain to hack on

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

fork of grain to hack on

License:Apache License 2.0

Languages

Language:Python 92.6%Language:Starlark 4.4%Language:C++ 1.7%Language:Shell 1.0%Language:Dockerfile 0.4%