LaQX: A JAX Toolkit for Neural Quantum State Simulations of Lattice Quantum Many-Body Models

Abstract

Neural quantum states provide accurate variational representations for interacting lattice fermions and spins. Turning a new ansatz into a reproducible calculation, however, requires coordinated implementations of the Hamiltonian, sampling, optimization, observables, and accelerator parallelism. We present LaQX (Lattice Quantum simulation based on jaX), an open-source neural quantum state library based on JAX that organizes these components through common functional interfaces. Beyond ground-state variational Monte Carlo, LaQX supports excited-state calculations, Green's Function Monte Carlo refinement, adiabatic evolution, and real-time evolution with time-dependent variational Monte Carlo. It consolidates methods developed in recent high-accuracy neural quantum state studies of the Hubbard model and frustrated magnets, and the package has been used to investigate the triangular-lattice Hofstadter-Hubbard model. We document these capabilities with reproducible calculations for Hubbard, J₁-J₂ Heisenberg, altermagnetic Hubbard, and interacting Hofstadter models. Function-oriented modules separate the scientific components and limit the source context needed for extensions by human developers and large language models. Source code, documentation, and executable examples are available at https://github.com/bytedance/laqx.

Time-Domain Observables for Resonant Scattering in Quantum Simulations of Lattice Gauge Theory
Praveen Balaji, Patrick Draper, Enrico Rinaldi
arXiv12 Oct 2026PreprintCC BY 4.0Full textHigh-energy & particle physics Quantum physics & quantum informationarXiv
From continuum neural quantum states to effective lattice models: Fluctuating Wigner molecules in moiré superlattices
Ahmed Khalifa, Shubhayu Chatterjee
arXiv6 Oct 2026PreprintCC BY 4.0Full textCondensed matterarXiv
clifford_qc: A Python Toolkit for Quantum Simulation
Ginanjar Utama, Hermawan Kresno Dipojono
arXiv18 Oct 2026PreprintarXiv licenceFull textQuantum physics & quantum informationarXiv
OpenULF: A Modular Toolkit Enables Data-Driven Ultra-Low-Field MRI with both Classical and Quantum Sensors
Hanhan Wu, Yuchen Yuan, Weiwen Jiang
arXiv1 Nov 2026PreprintarXiv licenceFull textBiological & medical physicsarXiv
Cavity mediated two-qubit gate: Tuning to optimal performance with NISQ era quantum simulations
Shreekanth S. Yuvarajan, Vincent Iglesias-Cardinale, David Hucul +1
Physical Review Research27 Oct 2026Peer-reviewedPublisher termsAtomic, molecular & optical physicsCrossref

Cite

BibTeX RIS

BibTeX

@misc{gu2026laqx,
  title = {{LaQX: A JAX Toolkit for Neural Quantum State Simulations of Lattice Quantum Many-Body Models}},
  author = {Yuntian Gu and Chaohui Fan and Ruichen Li and Haoxiang Chen and Weizhong Fu and Yixiao Chen and Dingshun Lv and Weiluo Ren and Tao Xiang and Liwei Wang and Mingpu Qin and Xiang Li},
  howpublished = {arXiv},
  year = {2026},
  month = nov,
  doi = {10.48550/arxiv.2611.04333},
  eprint = {2611.04333},
  archivePrefix = {arXiv},
  url = {https://arxiv.org/abs/2611.04333},
  note = {CC BY 4.0},
}

RIS

TY  - UNPB
TI  - LaQX: A JAX Toolkit for Neural Quantum State Simulations of Lattice Quantum Many-Body Models
AU  - Yuntian Gu
AU  - Chaohui Fan
AU  - Ruichen Li
AU  - Haoxiang Chen
AU  - Weizhong Fu
AU  - Yixiao Chen
AU  - Dingshun Lv
AU  - Weiluo Ren
AU  - Tao Xiang
AU  - Liwei Wang
AU  - Mingpu Qin
AU  - Xiang Li
PY  - 2026
DA  - 2026/11/03
PB  - arXiv
DO  - 10.48550/arxiv.2611.04333
UR  - https://arxiv.org/abs/2611.04333
AB  - Neural quantum states provide accurate variational representations for interacting lattice fermions and spins. Turning a new ansatz into a reproducible calculation, however, requires coordinated implementations of the Hamiltonian, sampling, optimization, observables, and accelerator parallelism. We present LaQX (Lattice Quantum simulation based on jaX), an open-source neural quantum state library based on JAX that organizes these components through common functional interfaces. Beyond ground-state variational Monte Carlo, LaQX supports excited-state calculations, Green's Function Monte Carlo refinement, adiabatic evolution, and real-time evolution with time-dependent variational Monte Carlo. It consolidates methods developed in recent high-accuracy neural quantum state studies of the Hubbard model and frustrated magnets, and the package has been used to investigate the triangular-lattice Hofstadter-Hubbard model. We document these capabilities with reproducible calculations for Hubbard, J_1-J_2 Heisenberg, altermagnetic Hubbard, and interacting Hofstadter models. Function-oriented modules separate the scientific components and limit the source context needed for extensions by human developers and large language models. Source code, documentation, and executable examples are available at https://github.com/bytedance/laqx.
ER  -

Abstract

Related

Cite