Source Code for "Physics-inspired transformer quantum states via latent imaginary-time evolution"

This repository contains the source code and implementation details for the manuscript: "Physics-inspired transformer quantum states via latent imaginary-time evolution" Overview: This code implements Physics-Inspired Transformer Quantum States (PITQS), a neural quantum state architecture based on the Latent Imaginary-Time Evolution (LITE) framework. It includes implementations for the frustrated square lattice $J_1$-$J_2$ Heisenberg model.