VMC

Variational Monte Carlo dynamics for tensor-network states (MPS/PEPS), implemented with JAX primitives and NetKet interfaces.

Features (for tensor-network states)

• Sequential sampling ¹
• Efficient energy evaluation
• Gauge removal for better stability ²
• Sliced/sparse gradients for balance between efficiency and memory use

Footnotes

1. W.-Y. Liu, Y.-Z. Huang, S.-S. Gong, and Z.-C. Gu. Accurate Simulation for Finite Projected Entangled Pair States in Two Dimensions. Physical Review B 103(23):235155, 2021. https://doi.org/10.1103/PhysRevB.103.235155 ↩

2. Y. Wu. Real-Time Dynamics in Two Dimensions with Tensor Network States via Time-Dependent Variational Monte Carlo. arXiv:2512.06768, 2025. https://doi.org/10.48550/arXiv.2512.06768 ↩