ENH Vectorize JSD code