A scalable sample-based quantum diagonalization solver for general Pauli-sum Hamiltonians
YUTARO IIYAMA✉, International Center for Elementary Particle Physics, The University of Tokyo, Japan
A scalable ground-state solver to be used in sample-based quantum diagonalization (SQD) is presented. The solver is built on JAX API, and is intended to be run on GPU platforms and possibly many-GPU high-performance computing facilities. It is capable of subspace diagonalization in up to
CCS Concepts: • Applied computing
Additional Key Words and Phrases: Subspace quantum diagonalization, Quantum-centric supercomputing, Open source
ACM Reference Format:
Yutaro Iiyama. 2026. A scalable sample-based quantum diagonalization solver for general Pauli-sum Hamiltonians. In Proceedings of Make sure to enter the correct conference title from your rights confirmation email (Conference acronym ’XX). ACM, New York, NY, USA, 10 pages. https://doi.org/XXXXXXX.XXXXXXX
1 Introduction
Sample-based quantum diagonalization (SQD) has emerged as one of the most promising application of near-term quantum computers[6, 9–12]. Regarded as a superior alternative to the variational quantum eigensolver[8], SQD has been applied to a wide variety of ground-state problems of quantum many-body systems, demonstrating energy approximation accuracies comparable to the best classical methods.
In SQD, a many-body Hamiltonian of an intractable dimensionality is projected onto a small subspace, in which the ground-state eigenpair of the reduced Hamiltonian is found. The projection subspace dimensions are identified through sampling quantum states realized on a quantum computater. In the original formulation, which is also known as quantum selected configuration interaction (QSCI), these states are the approximations of the ground state prepared with ansatz circuits. In the variant dubbed sample-based Krylov quantum diagonalization (SKQD), final states of time evolutions under the target Hamiltonian for multiple durations are sampled instead. Crucially, in all versions of SQD, the only role quantum computers play is to suggest the subspace dimensions. Everything else, including correcting potential errors in the obtained samples, is performed as classical computation. This separation of responsibility between quantum and classical computation, with a strong emphasis on the latter, is the innovation that allows SQD to be a candidate framework in which quantum advantage is achieved even with pre-fault-tolerant quantum computers.
It is of critical importance that “small” is a relative term in this context; in practice, the number of subspace dimensions can reach up to tens of billions. Even when the number of measurement outcomes from the quantum computer are less than a million, classical postprocessing can drastically enlarge the subspace size by e.g. identifying additional relevant dimensions through symmetry considerations. The implication is that the Hamiltonian projection and diagonalization is not performed sequentially, because it is impossible to store the reduced Hamiltonian matrix in memory unless it is known to be extremely sparse. Instead, a matrix-free method must be employed for diagonalization, in which the matrix-vector operation, which computes the result of multiplying a vector with the reduced Hamiltonian, must perform the projection in situ. Additionally, this computation would likely be executed in distributed parallel environments such as high-performance computing (HPC) facilities.
The central software components for large-scale SQD are therefore a matrix-free eigensolver and a specialized matrix-vector operation. Since both happen to involve many vector-vector and matrix-vector multiplications, they are amenable to GPU acceleration. In fact, as modern HPC architectures provide the majority of their compute through GPUs, it is essentially necessary that these components are implemented in a GPU-native manner.
There already exist at least two such implementations that are open source. The first is SBD-G[4], which is a header-only C++ library utilizing Thrust[3] and MPI to enable multi-GPU parallelism. This library specializes in SQD of chemistry and hard-core boson Hamiltonians, and has achieved state-of-the-art performance of diagonalizing a
The second implementation, which is more similar in scope and design to what is presented in this paper, is in Ref. [2] as a part of the demonstration of the functionalities of the CUDA Quantum library. Unlike SBD-G, the Hamiltonian it works with is expressed as a general Pauli sum of up to 64 qubits. The eigensolver is based on the thick-restarted Lanczos method, which requires retention of many large vectors in memory.
Our SQD solver, which is also open source, is written in Python and built on the JAX API[1]. Like the CUDA Quantum implementation, it targets general Pauli-sum Hamiltonians, but there is no limit to the number of qubits the Hamiltonians act on. The eigensolver employs the locally optimal conjugate gradient method, and is optimized for GPU memory footprint. By taking advantage of the built-in parallelization capabilities in JAX, the full solver can be run in many-GPU HPC as well as a single workstation, with minimal change in the user code.
We designed the library to make the eigensolver a standalone general utility which can be used to find the lowest few eigenpairs of any Hermitian operator, as long as its operation on a vector is implemented as a JAX just-in-time compiled function. This eigensolver can be scaled up to eigenvalue problems of
This paper presents our SQD solver with the goal of broadening its user base by clarifying its internal workings and demonstrating its capabilities. We describe the conjugate gradient eigensolver and matrix-vector operation in Section 2. The performance and scaling of the standalone eigensolver and the overall SQD solver are discussed in Section 3. Section 4 concludes with comments on the planned extensions.
2 SQD components
2.1 Eigensolver
The locally optimal conjugate gradient method for an eigenvalue problem
computes a series of solution estimates
This algorithm can be regarded as a single-vector unpreconditioned version of the better-known locally optimal block preconditioned conjugate gradient (LOBPCG) method, which can identify multiple lowest or highest eigenpairs of
When tackling a higher-dimensional problem or using a matrix-vector function that requires a large RAM allocation, the eigensolver can be executed in parallel computing environments by splitting the
It should be mentioned that we do also provide an option to compute the second and higher eigenpairs by passing to the eigensolver a list of vectors that the solution should be orthogonalized against. This crude approach introduces minimal additional memory footprint but is known to suffer from numerical instability, which is exactly why LOBPCG was invented. For SQD, this functionality is not used at any rate.
2.2 Reduced Hamiltonian matrix-vector operation
2.2.1 Algorithm.
We now discuss the SQD-specific matrix-vector operation function. First, we must clarify what exactly correspond to the
the subspace diagonalization of the
The eigenvalue problem to solve is therefore
which means that the matrix-vector operation must compute
The Hamiltonian
where each Pauli string
where
Looking at the matrix elements of the operators, because
Therefore, the projection of the Hamiltonian boils down to identifying a
There is at most one such index in
We use Algorithm 1 (X source index search) to find
Algorithm 1 X source index search. In the lines, StableSort refers to a sorting that keeps the original relative order of elements that have equal values, and
Require: Sorted list of binary indices
Ensure: List of indices
-
for all
do -
-
if then -
-
else -
-
end if -
end for
-
return
Plugging Equation (6)-(9) back into Equation (5), we get
where
In practice, the “take” function is used on
Because X source index search and the “take” function are both expensive operations, the latter especially when the
where
with
2.2.2 Caching static data.
The matrix-vector operation of Section 2.2.1 is called hundreds of times during the eigensolver execution in some cases. Only the
and the source indices
If each
2.2.3 Limitation on the number of subspace dimensions.
As evident from Algorithm 1, X source search is heavily reliant on stable sorting of subspace dimension indices. Since stable sorting of more than
However, global sorting of the binary indices is not a strict necessity. When the
3 Performance of the SQD solver
3.1 Eigensolver
To validate the conjugate gradient eigensolver, we compute the two lowest energy eigenvalues
where
Figure 1a shows the energy gap
Next, we verify that the eigensolver can run with distributed vectors, and investigate its time-to-solution scaling as the number of parallel nodes is increased. We consider the same TFIM Hamiltonian at

(a) 30-site TFIM energy gap as a function of

(b) Time for 32-site TFIM ground state energy determination.
Fig. 1. Validation of the conjugate gradient eigensolver
4 Conclusion
We presented a scalable, memory-efficient SQD solver for general Pauli Hamiltonians based on JAX. The solver consists of a general-purpose conjugate gradient eigensolver and an SQD-specific matrix-vector multiplication function that performs Hamiltonian projection internally. Both the eigensolver and the matrix-vector function support distributed vectors, allowing the overall solver to run in massively parallel GPU environments to leverage modern HPC architectures. The matrix-vector function can operate in several different caching modes, allowing control over the tradeoff between the execution time and memory footprint.
The SQD solver is currently limited to
Acknowledgments
This work was supported by Japan Society for the Promotion of Science KAKENHI Grant Numbers JP24K07042 and JP24H00689.
Code availability
The SQD solver code is available at https://github.com/UTokyo-ICEPP/rqutils/blob/main/rqutils/sqd.py with a corresponding documentation at https://rqutils.readthedocs.io/en/latest/apidoc/rqutils.sqd.html.
References
[1] James Bradbury, Roy Frostig, Peter Hawkins, Matthew James Johnson, Yash Katariya, Chris Leary, Dougal Maclaurin, George Necula, Adam Paszke, Jake VanderPlas, Skye Wanderman-Milne, and Qiao Zhang. 2018. JAX: composable transformations of Python+NumPy programs. http://github.com/jax-ml/jax
[2] NVIDIA Corporation. 2026. CUDA-Q Applications: Sample-Based Krylov Quantum Diagonalization (SKQD). https://nvidia.github.io/cuda-quantum/latest/applications/python/skqd.html.
[3] NVIDIA Corporation. 2026. Thrust. https://developer.nvidia.com/thrust.
[4] Jun Doi, Tomonori Shirakawa, Yukio Kawashima, Seiji Yunoki, and Hiroshi Horii. 2026. GPU-accelerated selected Basis Diagonalization with Thrust for SQD-based algorithms. arXiv:2601.16637 [cs.DC] doi:10.48550/arXiv.2601.16637
[5] Toshihiro Hanawa, Kengo Nakajima, Yohei Miki, Takashi Shimokawabe, Kazuya Yamazaki, Shinji Sumimoto, Osamu Tatebe, Taisuke Boku, Daisuke Takahashi, Akira Nukada, Norihisa Fujita, Ryohei Kobayashi, Hiroto Tadano, and Akira Naruse. 2024. Preliminary performance evaluation of Grace-hopper GH200. In 2024 IEEE International Conference on Cluster Computing Workshops (CLUSTER Workshops) (2024-09-24). IEEE, 184–185. doi:10.1109/clusterworkshops61563.2024.00050
[6] Keita Kanno, Masaya Kohda, Ryosuke Imai, Sho Koh, Kosuke Mitarai, Wataru Mizukami, and Yuya O Nakagawa. 2026. Quantum-selected configuration interaction: Classical diagonalization of Hamiltonians in subspaces selected by quantum computers. 8 (2026), 023268. Issue 2. doi:10.1103/dmn4-snfx
[7] Andrew V Knyazev. 1990. A preconditioned conjugate gradient method for eigenvalue problems and its implementation in a subspace. In Numerical Treatment of Eigenvalue Problems Vol. 5 / Numerische Behandlung von Eigenwertaufgaben Band 5: Workshop in oberwolfach, February 25 - march 3, 1990 / tagung in oberwolfach, 25. Februar - 3. März 1990 (1990-02-25) (International Series of Numerical Mathematics), J Albrecht, L Collatz, P Hagedorn, and W Velte (Eds.). Springer, 143–154. doi:10.1007/978-3-0348-6332-2
[8] Alberto Peruzzo, Jarrod McClean, Peter Shadbolt, Man-Hong Yung, Xiao-Qi Zhou, Peter J Love, Alán Aspuru-Guzik, and Jeremy L O’Brien. 2014. A variational eigenvalue solver on a photonic quantum processor. 5 (2014), 4213. Issue 1. doi:10.1038/ncomms5213
[9] Javier Robledo-Moreno, Mario Motta, Holger Haas, Ali Javadi-Abhari, Petar Jurcevic, William Kirby, Simon Martiel, Kunal Sharma, Sandeep Sharma, Tomonori Shirakawa, Iskandar Sitdikov, Rong-Yang Sun, Kevin J Sung, Maika Takita, Minh C Tran, Seiji Yunoki, and Antonio Mezzacapo. 2025. Chemistry beyond the scale of exact diagonalization on a quantum-centric supercomputer. 11 (2025), eadu9991. Issue 25. doi:10.1126/sciadv.adu9991
[10] Tomonori Shirakawa, Javier Robledo-Moreno, Toshinari Itoko, Vinay Tripathi, Kento Ueda, Yukio Kawashima, Lukas Broers, William Kirby, Himadri Pathak, Hanhee Paik, Miwako Tsuji, Yuetsu Kodama, Mitsuhisa Sato, Constantinos Evangelinos, Seetharami Seelam, Robert Walkup, Seiji Yunoki, Mario Motta, Petar Jurcevic, Hiroshi Horii, and Antonio Mezzacapo. 2025. Closed-loop calculations of electronic structure on a quantum processor and a classical supercomputer at full scale. arXiv:2511.00224 [quant-ph] doi:10.48550/arXiv.2511.00224
[11] Kenji Sugisaki, Shu Kanno, Toshinari Itoko, Rei Sakuma, and Naoki Yamamoto. 2025. Hamiltonian simulation-based quantum-selected configuration interaction for large-scale electronic structure calculations with a quantum computer. 27 (2025), 20869–20884. Issue 38. doi:10.1039/d5cp02202a
[12] Jeffery Yu, Javier Robledo Moreno, Joseph T Iosue, Luke Bertels, Daniel Claudino, Bryce Fuller, Peter Groszkowski, Travis S Humble, Petar Jurcevic, William Kirby, Thomas A Maier, Mario Motta, Bibek Pokharel, Alireza Seif, Amir Shehata, Kevin J Sung, Minh C Tran, Vinay Tripathi, Antonio Mezzacapo, and Kunal Sharma. 2025. Quantum-Centric Algorithm for Sample-Based Krylov Diagonalization. arXiv:2501.09702 [quant-ph]
A Locally-optimal conjugate gradient
Given a symmetric or Hermitian matrix
is an iterative procedure where the estimate solutions
with search (conjugate) directions
evaluated at
where
and
using some
In Equation (19), the step size
Therefore, the solution update at iteration
The right-hand side of Equation (25) is found by the Rayleigh-Ritz method.
Algorithm 2 is the pseudocode representation of our eigensolver implementation. It includes the details on the treatment of the first two iterations, which is ommitted in the discussion above.
Algorithm 2 Locally optimal conjugate gradient eigensolver
Require: Hermitian operator
Ensure: Lowest eigenvalue
-
function
-
for do -
for -
-
if and then -
-
end if -
end for -
if then -
Discard numerically unreliable direction -
end if -
return -
end function
-
function
-
for Hermitian -
lowest eigenpair of Analytic for / Cardano for -
return -
end function
-
if else -
-
-
-
-
-
-
-
while
and not do -
-
-
-
Orthogonal to new spans old -
-
-
-
-
-
Relative tolerance scale -
-
-
end while
-
return
Author’s Contact Information: Yutaro Iiyama, iiyama@icepp.s.u-tokyo.ac.jp, International Center for Elementary Particle Physics, The University of Tokyo, Bunkyo, Tokyo, Japan.
Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and /or a fee. Request permissions from permissions@acm.org.
© 2026 Copyright held by the owner/author(s). Publication rights licensed to ACM.
Manuscript submitted to ACM
About this conversion
“A scalable sample-based quantum diagonalization solver for general Pauli-sum Hamiltonians” by Yutaro Iiyama. Original arXiv submission, 2610.13907v1. Shared under Creative Commons Attribution 4.0 International, as identified by the exact-version arXiv record.
Linneon LLC converted the original LaTeX submission to continuous Markdown and HTML using LEC on October 14, 2026. Pagination and page furniture are removed; tables, notes and figures are arranged for continuous reading. HTML mathematics uses MathJax. These reading copies are not the official publication and do not imply endorsement by the authors, their institutions or arXiv. Original notices are retained. Dates generated by the source may reflect recompilation.