Capability
Mixed Precision Training With Automatic Loss Scaling
16 artifacts provide this capability.
Want a personalized recommendation?
Find the best match →Top Matches
1
imagen-pytorchFramework47/100
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