Also, follow up question: apex had levels e.g. ‘fp16_opt_level’: ‘O1’,  # For fp16: Apex AMP optimization level selected in [‘O0’, ‘O1’, ‘O2’, and ‘O3’]. Does torch.cuda.amp have anything similar I need to keep in mind?