Neural Variational Monte Carlo - Jax

Implementation of variational Monte Carlo in JAX, along with a simple neural network Ansatz, PsiMLP.

For more details please refer to my blog post, Neural Variational Monte Carlo.

Usage

Dependencies

Environment can be created very quickly with uv:

pip install uv
uv sync
source .venv/bin/activate

Blog

The file extra/blog/blog.py contains all the code to reproduce the original blog post, and can be run with:

cd extra/blog
python blog.py