Tutorial for GUD: Generation with Unified Diffusion.
A flexible library for Bijections and Normalizing Flows, powered by JAX. Designed for high-performance probabilistic modeling in physics contexts.