pytc.vmc.optimizer

Optimizer implementations for VMC.

Classes

NewtonOptimizer

Newton Optimizer (formerly Matrix-Free Optimizer).

Functions

apply_gradient_mask	Apply gradient mask to gradients to freeze parameters.
create_gradient_mask	Create a gradient mask PyTree for the combined params structure.
create_optimizer	Create an optimizer based on specified type and parameters.