diff --git a/bayesnewton/inference.py b/bayesnewton/inference.py index 60840aa..247a955 100644 --- a/bayesnewton/inference.py +++ b/bayesnewton/inference.py @@ -8,7 +8,8 @@ inv_vmap, vmap_diag, solve, - ensure_diagonal_positive_precision + ensure_diagonal_positive_precision, + mvn_logpdf_ ) from .likelihoods import Likelihood, MultiLatentLikelihood from .basemodels import GaussianDistribution, SparseGP