Pax: A Jax-based machine learning framework for training large scale models | Hacker News Reader