oml
val normal_kl_divergence : p_mean:float -> p_sigma:float -> q_mean:float -> q_sigma:float -> float