Geek Repo
followers
following
stars
Github PK Tool:Github PK Tool
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.
A simple, performant and scalable Jax LLM!