add KL divergence estimator to np_divs