megatron.optimizer.grad\_scaler.MegatronGradScaler
==================================================

.. currentmodule:: megatron.optimizer.grad_scaler

.. autoclass:: MegatronGradScaler
   :members:
   :special-members:
   :show-inheritance:
   :exclude-members: __weakref__, __init__