A unified framework for diffusion models, with applications specifically tailored for Lattice Field Theory. Implements state-of-the-art generative modeling techniques.
View on GitHubA flexible library for Bijections and Normalizing Flows, powered by JAX. Designed for high-performance probabilistic modeling in physics contexts.
View on GitHub