MCMC and Bayesian models.

https://en.wikipedia.org/wiki/Probabilistic_programming

They're trying to drum up interest in Jax. They do this time to time, but yeah, good for them doing what they like.