English

Qiskit-Torch-Module: Fast Prototyping of Quantum Neural Networks

Quantum Physics 2025-01-15 v2 Machine Learning Software Engineering

Abstract

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.

Keywords

Cite

@article{arxiv.2404.06314,
  title  = {Qiskit-Torch-Module: Fast Prototyping of Quantum Neural Networks},
  author = {Nico Meyer and Christian Ufrecht and Maniraman Periyasamy and Axel Plinge and Christopher Mutschler and Daniel D. Scherer and Andreas Maier},
  journal= {arXiv preprint arXiv:2404.06314},
  year   = {2025}
}

Comments

Accepted to the IEEE International Conference on Quantum Computing and Engineering (QCE 2024), Montr\'eal, Qu\'ebec, Canada. 7 pages, 4 figures, 3 tables