netket.models.JastrowΒΆ

class netket.models.Jastrow(dtype=<class 'jax._src.numpy.lax_numpy.complex128'>, kernel_init=<function normal.<locals>.init>, parent=<flax.linen.module._Sentinel object>, name=None)[source]ΒΆ

Bases: flax.linen.module.Module

Jastrow wave function \(\Psi(s) = \exp(\sum_{ij} s_i W_{ij} s_j)\).

The W matrix is stored as a non-symmetric matrix, and symmetrized during computation by doing W = W + W.T in the computation.

Attributes
variablesΒΆ

Returns the variables in this module.

Return type

Mapping[str, Mapping[str, Any]]

Methods
kernel_init(shape, dtype=<class 'jax._src.numpy.lax_numpy.float64'>)ΒΆ
put_variable(col, name, value)ΒΆ

Sets the value of a Variable.

Parameters
  • col (str) – the variable collection.

  • name (str) – the name of the variable.

  • value (Any) – the new value of the variable.