Will Cromar

It looks like there are two issues in your example notebook. First, the torch version isn't updated to 2.2 yet. I just sent a PR to do this: https://github.com/Kaggle/docker-python/pull/1364 Kaggle...

Thanks for the heads up. I'll work with the Kaggle team to get that image updated.

@miladm Are you testing with `PJRT_DEVICE=CPU`? > Turns out the wrong shape size function is called by ignoring dynamism information propagated downstream. Could [this comment](https://github.com/pytorch/xla/pull/4077#discussion_r992531413) be related? It wasn't clear...

- Original PRs: #7249 and #7268 - Reason to backport: Enables new libtpu telemetry and detection of CUDA plugin package (`torch_xla_cuda_plugin`) by default. - Backport link: #7270

Hey @ysiraichi, I'll spend some more time going over this PR tomorrow to try to understand it better. We were just preparing to remove the separate GPU variant of the...