https://github.com/markusschmitt/vmc_jax
This is an impementation of Variational Monte Carlo (VMC) for quantum many-body dynamics using the JAX library (and Flax on top) to exploit the blessings of automatic differentiation for easy model composition and just-in-time compilation for execution on accelerators.
Please report bugs as well as other issues or suggestions on our issues page.
Documentation is available here.
pipinstall pip install jVMC
To install jVMC with GPU support, use