Add Jastrow model (and some cleanup) (!577) · Merge requests · Vicentini Filippo / netket

Vicentini Filippo requested to merge jastrow-jax into master Mar 23, 2021

Created by: femtobit

For old times sake, this PR adds netket.models.Jastrow with test and example for Ising 1D.

There is also some cleanup in this PR, in particular the type aliases defined in each file in models are now in one location netket.utils.types.

Also, I've slightly extended EarlyStopping for the example, so it is possible to stop when an exact eigenstate is reached (through variance ≈ 0).

Add Jastrow model (and some cleanup)