Implementation of Variational Monte Carlo with JAX. The files contain the following:

• energy.py : functions to compute energy
• energy_minimisation.py : functions to minimise energy
• sampling.py : functions to perform Metropolis-Hastings algorithm
• time_autocorrelation.py : functions to compute auto-correlation time
• neural_network.py : classes for Neural Quantum States (i.e. Neural Network wavefunctions)