Home / JAX / New Libraries > Pax New Libraries > Pax A Jax-based machine learning framework for training large scale models. Package 550 stars GitHub Back to JAX