xla

73c2fd06 - Significantly reduce Docker image size (#4044)

Significantly reduce Docker image size (#4044) * Create new multi-stage docker build * WIP GPU support * Make CUDA optional * Support patches, install torchvision * Fix patching * Address review comments