Speed up Qubit Research with NVIDIA cuQuantum Integrations in QuTiP and scQubits

NVIDIA cuQuantum is an SDK of libraries for accelerating quantum simulations on the circuit (digital) and device (analog) level. It’s now integrated into Quantum Toolbox in Python (QuTiP) and Superconducting Qubits (scQubits), enabling end-to-end acceleration in workflows for designing and studying novel sorts of qubits.

QuTiP is a widely used package for simulating the time evolution of open quantum systems. scQubits is the most well-liked package for modeling superconducting qubits. Together, these packages enable users to define the parameters of novel qubit systems and evolve them in response to manage pulses. Users can study their interactions with components corresponding to filters or resonators. They may calculate critical system parameters, corresponding to frequency shifts and transition energies. This permits rapid prototyping of novel device designs to enhance quantum device performance.

Researchers in Alexandre Blais’ group on the University of Sherbrooke led the way in which, developing and integrating a brand new cuQuantum plugin for QuTiP, qutip-cuquantum. The technology delivers a 4000x speedup from CPU to an 8x GPU node hosted on AWS for a big transmon-resonator system. The QuTiP community can use the plugin to realize top performance at scale, which results in less noisy systems, addressing considered one of the important thing bottlenecks to scaling useful quantum computers.

The qutip-cuquantum plugin also supports the scaling of simulations to much larger Hilbert spaces with multi-GPU and multi-node capabilities, enabling the study of more complex quantum systems. Blais’ group also scaled to much larger systems (64-state transmon qubit / 512-state resonator pair), with the assistance of the P5en instance on AWS, simulations that were too large to run without multi-GPU support.

When running QuTiP-cuQuantum mesolve on AWS P5en instances equipped with NVIDIA B200 GPUs, users could see a 4000x speedup over running QuTiP with Hpc7a CPU instances for a Hilbert space with a 32-level qudit and a 512-level resonator. — *Figure 1. Results for the QuTiP master equation solver on CPU and GPU accelerated by NVIDIA cuQuantum*

Anyone considering QuTiP simulations at scale can now use cuQuantum on AWS for nice performance in quantum dynamics simulation workloads. This permits QPU designers and researchers to know how more complex dynamics affect their qubit designs.

Jens Koch’s group at Northwestern University is the developer of scQubits, an open source Python package for modeling superconducting qubits. Working with scQubits, NVIDIA developed APIs in cuQuantum that speed up this package for executing the critical parts of the total qubit design workflow on NVIDIA GPUs.

Eigensolvers enable the computation of the energy spectra of superconducting devices, a very important quantity vital to design recent sorts of qubits. scQubits provides an easy way for researchers to compute energy spectra given the physical parameters of a superconducting circuit, for instance, the capacitance and inductance of varied circuit components.

The chart shows that using scQubits with cuQuantum results in a 54x speedup when solving for the eigenvalues of a superconducting quantum system. — Figure 2. scQubits with cuQuantum running on NVIDIA DGX B200 GPUs, shows a 54x speedup over advanced multi-threaded data center Intel Emerald Rapids CPUs for a system composed of 5 4-level fluxonium qubits and 4 8-level resonators

The outputs of scQubits may easily function inputs for analog quantum dynamics simulations using QuTiP-cuQuantum. Designers can develop recent quantum devices with improved coherence times, gate and readout performances, greater throughput, and fewer limitations on accessible system sizes through the use of GPU acceleration of scQubits and QuTiP with NVIDIA hardware and software.

A graph showing that running the corresponding fluxonium resonator system from scQubits in QuTiP-cuQuantum on NVIDIA DGX B200 GPUs enables a 135x speedup for a 40-level fluxonium and 25-level resonator mixed-state simulations compared to an 8570 Emerald Rapids CPU. — Figure 3. Taking the output from scQubits and running a mixed-state fluxonium resonator system in QuTip-cuQuantum enables 135x speedup from an 8570 Emerald Rapids data center CPU in comparison with an NVIDIA DGX B200 GPU, along with with the ability to scale much further

With multi-GPU multi-node execution in each tools, Koch’s group can explore more complex composite qubit systems beyond just the single-qubit unit cell of a fluxonium (or other circuit) and a resonator.

Other developers will soon have the opportunity to scale their scQubits and QuTiP simulations as much as previously out-of-reach lattices of multiple degrees of freedom to know how multiple qubit systems interact with one another.

Download the GPU-accelerated QuTiP to enable your workloads on NVIDIA hardware by installing from PyPI with pip install qutip-cuquantum.

scQubits will soon support cuQuantum, enabling greater than an order of magnitude speedup over CPU and scaling to regimes previously intractable.

cuQuantum 25.09 is out and enables these workflows with best-in-class performance and scale, ensuring that quantum device designers are getting the proper answers faster than before, reducing the timeline to useful quantum computers. Download the cuQuantum SDK. The most recent packages might be found on PyPI or Conda, and might be installed by pip install cuquantum-python-cu13.

Source link