Wrong name in APEX optimization levels
APEX optimization levels are “O1, O2, O3” and not “01, 02, 03”. We should fix this in code + docs.

Make sure it is clear from the docs that the use of Apex anymore and recommend that users use upstream native AMP available since PyTorch 1.6