You could profile the code to further isolate the bottleneck of the script, which could help in further debugging. E.g. we’ve been working on the usage of cusolver in more torch.linalg methods, which could speed up your workflow in case you are using the nightly binaries or a source build.