Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
AlphaUncommon

google/paxml

Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.

Owner
google (organization)
Stars
Stars: 560
Forks
Forks: 73
Language
Python
License
Apache-2.0
Created
2022
Topics
c4jaxlarge-language-modelsllmmodel-flopsparallelismgpt
Tier
Alpha (250 to 699 ★) What this tier means
View on GitHub

More repos by google

All 1,050 repos by google

Nearby in Python

All Python repos