georgiyekkert / paxml Goto Github PK
View Code? Open in Web Editor NEWThis project forked from 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.
License: Apache License 2.0