# Automatic Mixed Precision package - tensorplay.amp

Source: https://www.tensorplay.cn/docs/amp.html

# Automatic Mixed Precision package - tensorplay.amp

## Autocasting

[tensorplay.amp.autocast_mode.is_autocast_available](generated/tensorplay.amp.autocast_mode.is_autocast_available.html#tensorplay.amp.autocast_mode.is_autocast_available)

[tensorplay.amp.autocast_mode.autocast](generated/tensorplay.amp.autocast_mode.autocast.html#tensorplay.amp.autocast_mode.autocast)

Instances of autocast serve as context managers or decorators that allow regions of your script to run in mixed precision.

[tensorplay.amp.autocast_mode.custom_fwd](generated/tensorplay.amp.autocast_mode.custom_fwd.html#tensorplay.amp.autocast_mode.custom_fwd)

Create a helper decorator for forward methods of custom autograd functions.

[tensorplay.amp.autocast_mode.custom_bwd](generated/tensorplay.amp.autocast_mode.custom_bwd.html#tensorplay.amp.autocast_mode.custom_bwd)

Create a helper decorator for backward methods of custom autograd functions.

## Gradient Scaling

[tensorplay.amp.grad_scaler.GradScaler](generated/tensorplay.GradScaler.html#tensorplay.GradScaler)

An instance scaler of GradScaler.

[tensorplay.amp.is_autocast_available](generated/tensorplay.amp.is_autocast_available.html#tensorplay.amp.is_autocast_available)