xla

Remove patch which landed and downgrade torchvision

#371

Merged