pytc.vmc.optimization.make_second_order_training_step

pytc.vmc.optimization.make_second_order_training_step(mcmc_step, optimizer, n_mcmc_per_opt=1, n_opt_per_mcmc=1)[source]

Factory to create unified training step for optimizers with a .step() method.

This supports Newton-style optimizers which handle gradient computation internally via their step() method.