netket.optimizer.qgt.QGTOnTheFly#
- netket.optimizer.qgt.QGTOnTheFly(vstate, *, chunk_size=None, holomorphic=None, **kwargs)[source]#
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