steventk-g

Here's a link to the virtual device draft PR, it's still an untested work-in-progress https://github.com/pytorch/xla/pull/4091

After https://github.com/pytorch/xla/pull/3843, we will need changes to support (1) einsum on more than 2 inputs and (2) einsum on equations like `ijj,k->ik`, where one input or output has an index...

From discussing with Blake, there are some changes we can make to fix problem 2: - Put ones in the shape instead of 0s for the missing index (`j` in...

Addressed the issue with https://github.com/pytorch/xla/pull/4333

@reciprocated feel free to assign this me, just wanted to put it on your radar.