pytorch/torch/cpu/amp
..
__init__.py
autocast_mode.py
grad_scaler.py