Training loop of PyTorch without boilerplate code
conda install esri::accelerate
conda install esri/label/prerelease::accelerate