Support Automatic Mixed Precision (AMP)#51

Open

Queuecumber wants to merge 1 commit intofbcotter:masterfrom

Queuecumber:master

Commits on Aug 2, 2023

Correctly annotate autograd functions to support amp
Queuecumber
committed