netket.models.Jastrow#

class netket.models.Jastrow[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
has_rng(name)#

Returns true if a PRNGSequence with name name exists.

Return type

bool

Parameters

name (str) –

kernel_init(shape, dtype=<class 'jax.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.

Returns: