That’s good to hear, as more recent releases ship with the latest bug fixes.
Note that apex.amp is deprecated (in case you are using this utility from apex) and you should use the native implementation via torch.cuda.amp.