paxml
Python
★ 556
updated 16d ago
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.
No plain-English explanation yet — one is being written right now. Check back in a minute.