Capability
Automatic Mixed Precision Training With Precision Plugins
13 artifacts provide this capability.
Want a personalized recommendation?
Find the best match →Top Matches
1
imagen-pytorchFramework47/100
via “mixed precision training with automatic loss scaling”
Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
Unique: Integrates Accelerate's mixed precision with automatic loss scaling, handling precision casting and numerical stability without manual configuration
vs others: Provides automatic mixed precision with loss scaling through Accelerate, reducing boilerplate compared to manual precision management while maintaining numerical stability