over the first anneal_rho steps.
gamma = lerp(global_step, 0, aneal_rho,
rho_to_gamma(100.0), gamma)