Searcharxiv⌕ Search
Home/Archive/arXiv

arXiv · 2611.04333

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

Yuntian Gu·Chaohui Fan·Ruichen Li·Haoxiang Chen·Weizhong Fu·Yixiao Chen·Dingshun Lv·Weiluo Ren·Tao Xiang·Liwei Wang·Mingpu Qin·Xiang Li
View original source ↗Read source document ↗Export metadata ↓

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_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.

Explore related subjects

cond-mat.str-el

Keep this discovery

Explore connections, maps & timelines

BibTeXRIS

Yuntian Gu, Chaohui Fan, Ruichen Li, Haoxiang Chen, Weizhong Fu, Yixiao Chen, Dingshun Lv, Weiluo Ren, Tao Xiang, Liwei Wang, Mingpu Qin, Xiang Li. 2026-11-03. LaQX: A JAX Toolkit for Neural Quantum State Simulations of Lattice Quantum Many-Body Models. https://arxiv.org/abs/2611.04333

Cite the original work for its findings. Save a collection to share your selection of sources.

KEEP EXPLORING

Related papers

arXiv2026

An Atlas of Classical Pyrochlore Spin Liquids

Frustrated magnetism in the pyrochlore lattice magnet has proven to be a most fruitful setting for the experimental and theoretical search for spin liquids. Besides the canonical case of spin ice, recent works have identified a variety of new classical and quantum spin liquids engendered by the generic nearest-neighbor anisotropic spin Hamiltonian for that lattice. However, a general framework for the thorough and systematic classification and characterization of these exotic states of matter has been lacking, as has an exhaustive list of all possible spin liquids that this model can support and, perhaps most interesting, what is the corresponding structure of their emergent field theory description. In this work, we develop such a theoretical framework to identify the interaction parameters stabilizing different classical spin liquids and derive their corresponding effective generalized Gauss's laws at low temperatures. Combining this with Monte Carlo simulations, we systematically identify all classical spin liquids for the general nearest-neighbor anisotropic spin Hamiltonian on the pyrochlore lattice. In doing so, we uncover new spin liquid models with exotic forms of generalized Gauss's law and multipole conservation laws. Our approach allows us to compile an atlas of all spin liquids realized in the phase diagram, providing a global picture of their mutual connections in parameter space and transitions between them. Our work will inform future theoretical and experimental studies of classical and quantum spin liquids on the pyrochlore lattice and help rationalize the exotic properties of pyrochlore magnets.

cond-mat.str-el↗
arXiv2026

Spectroscopic Evidence of Competing Diagonal Spin Interactions and Spin Disproportionation in the Bilayer Nickelate La$_3$Ni$_2$O$_7$

A comprehensive spectroscopic map of the electronic, magnetic, and lattice excitations is presented for the bilayer nickelate La$_3$Ni$_2$O$_7$ using Raman scattering at ambient pressure. Upon entering the spin density wave state below 153 K, the $A_{1g}$ channel exhibits an abrupt electronic spectral gap with a clear isosbestic point. In contrast, the $B_{1g}$ and $B_{2g}$ channels are dominated by pronounced two-magnon (2M) excitations, representing a signature of incipient Mottness. These 2M signals in both channels constitute evidence for two distinct in-plane spin exchange interactions along the Ni-O bonding and its diagonal directions. Calculations based on the spin wave theory further reveal that the 2M mode in the $B_{2g}$ channel arises from the competition between two bond-diagonal antiferromagnetic interactions mediated by nickel $d_{x^2-y^2}$ orbitals. Furthermore, emergent low-energy 2M excitations below 10 meV are found to originate from distinct, weaker spin moments, supporting spin disproportionation. Simultaneously, an anomalous softening of $B_{1g}$ phonons from 280 down to 4.5 K is uncovered, suggesting the presence of an incipient lattice instability leading to checkerboard-type breathing modulations. Collectively, these findings identify a ground state of the bilayer nickelate characterized by competing bond-diagonal interactions, spin disproportionation, and an incipient lattice instability, establishing key ingredients for understanding the mechanism of nickelate superconductivity.

cond-mat.str-el↗
arXiv2026

Direction-selective intertwined charge, orbital, and lattice orders under uniaxial strain in hole-doped manganite: La0.75Ca0.25MnO3

The complex interplay of charge, spin, orbital, and lattice degrees of freedom governs emergent phases in quantum materials, making strain a powerful control parameter. Recent advances in free-standing layer techniques have enabled extreme strains of nearly 8%, opening access to novel and often unexpected electronic and magnetic phases. Here, using a density functional theory approach, we investigate the effect of direction-selective uniaxial strain on the prototypical Jahn-Teller system La1-xCaxMnO3 (x = 0.25). We find that different strain directions stabilize qualitatively distinct structural, charge, and orbital responses, rather than merely different strengths of the same phase. In particular, extreme uniaxial strain selectively induces cooperative Jahn-Teller, breathing-like, and site-selective modulations, thereby enabling previously inaccessible intertwined orders in manganites. These results establish direction-selective uniaxial strain as a powerful and selective route for engineering emergent phases in quantum materials.

cond-mat.str-el↗