https://github.com/MathPhysSim/FERMI_RL_Paper/blob/cd2545c5146363e7966f86378d847177444b6c74/naf2_new.py#L564-L569
FERMI_RL_Paper/naf2_new.py
Lines 564 to 569 in cd2545c