AMP support #5
Description
Is your feature request related to a problem? Please describe.
Mixed precision is now built-into PyTorch since 1.6.0 [ref]
Describe the solution you'd like
Add this to the training/inference process to make things more efficient.
Additional context
Lock PyTorch version to >= 1.6.0