MP-NQS Implementation of the MP-NQS in JAX. Requirements are NetKet version 3.4 or larger. The file electrongas.py contains an example usage of the MP-NQS for the HEG with 14 electrons.