Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX. - View it on GitHub