apex.amp is depredcated in favor of the native mixed-precision utility used via torch.cuda.amp. Take a look at the examples 3 to see its usage.
The linked package is also not official and thus also doesn’t come with pre-built C++/CUDA kernels.