Pax: A Jax-based machine learning framework for training large scale models #1 Post by spallas » Wed, Aug 30, 2023, 9:44 AM UTC Pax: A Jax-based machine learning framework for training large scale modelsgithub.com