Path	Lines of Code
torch_xla/amp/__init__.py	2
torch_xla/amp/autocast_mode.py	4
torch_xla/amp/grad_scaler.py	55
torch_xla/amp/syncfree/__init__.py	1
torch_xla/amp/syncfree/sgd.py	70