netket.optimizer.qgt.QGTOnTheFly

netket.optimizer.qgt.QGTOnTheFly(vstate=None, *, chunk_size=None, holomorphic=None, **kwargs)

Lazy representation of an S Matrix computed by performing 2 jvp and 1 vjp products, using the variational state’s model, the samples that have already been computed, and the vector.

The S matrix is not computed yet, but can be computed by calling to_dense. The details on how the ⟨S⟩⁻¹⟨F⟩ system is solved are contained in the field sr.

Parameters:

• vstate – The variational State.

• chunk_size – If supplied, overrides the chunk size of the variational state (useful for models where the backward pass requires more memory than the forward pass).

• holomorphic (bool | None)

Return type:

QGTOnTheFlyT