Scale JAX models to multi-GPU systems
Google Cloud Tech · 3:28
Ce court tutoriel montre que passer d'un entraînement JAX mono-GPU à multi-GPU relève avant tout du placement des tableaux (sharding) et non d'une réécriture du code d'entraînement, puis applique ces briques à un « ti...