Star AlbumentationsX on GitHub — it powers this leaderboard
jax-ml/bayeux
State of the art inference for your bayesian models.