Writing a simple probabilistic programming language using interpreters in JAX. - View it on GitHub