jax.scipy.stats.gaussian_kde.integrate_gaussian

gaussian_kde.integrate_gaussian(mean, cov)

对高斯分布加权后的分布进行积分。