enable tensorsolve - Githubissues

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)

https://pytorch.org/xla

Other

2.49k stars 480 forks source link

enable tensorsolve #8287

Closed barney-s closed 3 weeks ago

barney-s commented 1 month ago

Enable tensorsolve

use jnp tensorsolve
handle moving dims/axes and reshaping b before calling jnp variant

Fixes: #7505