Probabilistic programming with NumPy powered by JAX for autograd and JIT...
Manifold Markov chain Monte Carlo methods in Python