Torch.cuda.amp vs Nvidia apex?

[mcarilli (Michael Carilli)]

Yes, the action of those hooks should be composable with the things amp does.