⭐ Star AlbumentationsX on GitHub — 535+ stars and counting!
jax-ml/bayeux
State of the art inference for your bayesian models.