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 trainable-leaf mask for the combined parameter PyTree.
create_optimizer	Create an optimizer based on specified type and parameters.