Project Awesome project awesome

New Libraries > Pax

A Jax-based machine learning framework for training large scale models.

Package 550 stars GitHub
Back to JAX