[jdeaton]

if you're using tpu why are you using pytorch

[hustwindmaple1]

there is limited TPU support in pytorch via torch_xla

[jdeaton]

Sounds limited