Skip to content

darsnack/grain

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

264 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

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.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 93.8%
  • Starlark 3.6%
  • C++ 1.5%
  • Other 1.1%