Skip to content

cqsl/sparse-twobody-arnn

Repository files navigation

Sparse Autoregressive Neural Networks for Classical Spin Systems

Paper link: arXiv:2402.16579 | Mach. Learn.: Sci. Technol. 5 025074 (2024)

The code uses Equinox to define neural networks.

main.py runs the VMC training of TwoBo or MADE, or the exact enumeration, given a Hamiltonian instance. args.py contains all the configurations.

reproduce.sh contains the commands to reproduce the TwoBo and MADE results in the paper. In practice, you may run these commands in parallel on multiple GPUs, and set your output directory.