VAN-DAMME: GPU-accelerated and symmetry-assisted quantum optimal control of multi-qubit systems
We present an open-source software package, VAN-DAMME (Versatile Approaches to Numerically Design, Accelerate, and Manipulate Magnetic Excitations), for massively-parallelized quantum optimal control (QOC) calculations of multi-qubit systems. To enable large QOC calculations, the VAN-DAMME software package utilizes symmetry-based techniques with custom GPU-enhanced algorithms. This combined approach allows for the simultaneous computation of hundreds of matrix exponential propagators that efficiently leverage the intra-GPU parallelism found in high-performance GPUs. In addition, to maximize the computational efficiency of the VAN-DAMME code, we carried out several extensive tests on data layout, computational complexity, memory requirements, and performance. These extensive analyses allowed us to develop computationally efficient approaches for evaluating complex-valued matrix exponential propagators based on Padé approximants. To assess the computational performance of our GPU-accelerated VAN-DAMME code, we carried out QOC calculations of systems containing 10 - 15 qubits, which showed that our GPU implementation is 18.4× faster than the corresponding CPU implementation. Our GPU-accelerated enhancements allow efficient calculations of multi-qubit systems, which can be used for the efficient implementation of QOC applications across multiple domains. Program Title: VAN-DAMME CPC Library link to program files:: https://doi.org/10.17632/zcgw2n5bjf.1 Licensing provisions: GNU General Public License 3 Programming language: C++ and CUDA Nature of problem: The VAN-DAMME software package utilizes GPU-accelerated routines and new algorithmic improvements to compute optimized time-dependent magnetic fields that can drive a system from a known initial qubit configuration to a specified target state with a large (≈1) transition probability. Solution method: Quantum control, GPU acceleration, analytic gradients, matrix exponential, and gradient ascent optimization.
Authors
- José Rodríguez-Borbón
- Bryan M. Wong (ORCID: https://orcid.org/0000-0002-3477-8043)
- Adrián Perez Diéguez (ORCID: https://orcid.org/0000-0001-7168-9050)
- Khaled Z. Ibrahim (ORCID: https://orcid.org/0009-0004-5362-3612)
- Xian Wang (ORCID: https://orcid.org/0000-0002-1802-6925)
Publication Details
- Journal
- Computer Physics Communications
- Published
- 2024-10-17
- DOI
- https://doi.org/10.1016/j.cpc.2024.109403
- Citations
- 3
- Primary Topic
- Quantum Information and Cryptography
- Type
- article
- Field-Weighted Citation Impact
- 1.04
Funders
- U.S. Department of Energy
- National Energy Research Scientific Computing Center
- Office of Science
- Advanced Scientific Computing Research
- Lawrence Berkeley National Laboratory