[Feature] Optimize Krylov solver for batched calculations

pasqal-io / pyqtorch

PyTorch-based state vector simulator

https://pasqal-io.github.io/pyqtorch/

Apache License 2.0

45 stars 15 forks source link

[Feature] Optimize Krylov solver for batched calculations #295

Open vytautas-a opened 2 weeks ago

vytautas-a commented 2 weeks ago

Describe the feature

Currently (#294) batching in Krylov solver is implemented straightforwardly using a simple loop. This should be optimized to harness the power of broadcasted tensor operations.

It should be implemented because

No response

Additional context

No response

Would you like to work on this issue?

None