Optax

Optax sert de bibliothèque de traitement et d'optimisation des gradients spécifiquement conçue pour JAX, facilitant la recherche en fournissant des blocs de construction modulaires qui peuvent être facilement combinés en optimisateurs personnalisés. Elle vise à améliorer la productivité de la recherche avec des implémentations efficaces de composants essentiels, permettant aux utilisateurs de créer des solutions sur mesure. La bibliothèque comprend divers optimisateurs et fonctions de perte populaires, ce qui la rend adaptée aux applications d'apprentissage automatique.