* relaxing the finite differences tolerance
 * trying to reduce the overhead of compute_psis (training bottleneck)