pytc.vmc.test.test_optimizer_mask

Tests for freezing selected Jastrow parameters during optimization.

Classes

FrozenJastrow
TestGradientMask
TestPublicOptimizeFrozenParams	Public-path regression for the original production defect: if the gradient_mask wiring inside optimize() is removed, the Optax path silently trains frozen leaves — and until this test existed, every other test in this file still passed.
TrainableJastrow

Functions

_ansatz
_params