jax.scipy.stats.gaussian_kde.logpdf