In CUDA/Apex AMP, you set the optimization level:
model, optimizer = amp.initialize(model, optimizer, opt_level="O1")
In the examples I read on PyTorch’s website, I don’t see anything analogous to this.  How is this accomplished?