4 ms·Pax: A Jax-based machine learning framework for training large scale models2 points by spallas 3y ago