maxtext
Pythonpushed almost 2 years ago
A simple, performant and scalable Jax LLM!
AI summary
LLM framework
A high-performance LLM written in Python/Jax for training and inference on Google Cloud TPUs and GPUs.
- stars
- 1.6K
- forks
- 304
- watching
- 40
- awesome lists
- 2
Featured in 2 awesome lists
Each link jumps to the spot where the list mentions maxtext.