- netket.optimizer.qgt.QGTOnTheFly(vstate=None, *, chunk_size=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.
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).
- Return type: