google/paxml

Pax est un framework de machine learning basé sur Jax pour l'entraînement de modèles à grande échelle. Pax permet une expérimentation et une parallélisation avancées et entièrement configurables, et a démontré des taux d'utilisation des FLOPs de modèle parmi les meilleurs du secteur.