GitHub repository: mamarder/ramsey
Description: Probabilistic modelling using Haiku and JAX