🎲🔥 Unofficial implementation of Newtonian Monte Carlo MCMC sampling algorithm (after Arora et al., 2020) in JAX/NumPyro, with full NumPyro-API compatibility - View it on GitHub