Qiskit-Torch-Module: Fast Prototyping of Quantum Neural Networks
This work addresses performance bottlenecks for researchers using Qiskit in day-to-day quantum computing simulations, though it is incremental as it builds on existing software.
The authors tackled the inefficiency of quantum computer simulation frameworks for training variational quantum algorithms by developing the qiskit-torch-module, which improves runtime performance by two orders of magnitude over comparable libraries and integrates quantum neural networks with PyTorch.
Quantum computer simulation software is an integral tool for the research efforts in the quantum computing community. An important aspect is the efficiency of respective frameworks, especially for training variational quantum algorithms. Focusing on the widely used Qiskit software environment, we develop the qiskit-torch-module. It improves runtime performance by two orders of magnitude over comparable libraries, while facilitating low-overhead integration with existing codebases. Moreover, the framework provides advanced tools for integrating quantum neural networks with PyTorch. The pipeline is tailored for single-machine compute systems, which constitute a widely employed setup in day-to-day research efforts.