xla

9d5f6f68 - Index device error (#3632)

Index device error (#3632) * add torch pin * tensor indexing: error when running cpu_tensor[xla_tensor] * lint * lint2 * lint (again) * fix * fix * fix