Official repository for “DeCo: Frequency-Decoupled Pixel Diffusion for End-to-End Image Generation”

Latent Diffusion Models with Masked AutoEncoders (LDMAE) official code

Official inference repo for FLUX.2 models

Official implementation for SSDD Single-Step Diffusion Decoder for Efficient Image Tokenization.

Official PyTorch implementation of FlowMo.

Official PyTorch Implementation of "Latent Denoising Makes Good Visual Tokenizers"

Benchmarking of machine learning and numerical weather prediction (MLWP & NWP) models, with a focus on extreme events.

PyTorch implementation of JiT https://arxiv.org/abs/2511.13720

Official Code of the paper “Equivariant Sampling for Improving Diffusion Model-based Image Restoration“

PyTorch Image Quality Assessement package

Diffusion models in PyTorch

Official Pytorch Implementation for "MultiDiffusion: Fusing Diffusion Paths for Controlled Image Generation" presenting "MultiDiffusion" (ICML 2023)

[CVPR 2025 Oral] Improving Diffusion Inverse Problem Solving with Decoupled Noise Annealing

Vite & Vue powered static site generator.

Learning in infinite dimension with neural operators.

Earth system foundation model data, training, and eval

We read tacos 📖

We make tacos 🔨

JAX + Equinox implementation for Neural Stochastic Flows.

SLAMS: Score-based Latent Assimilation in Multimodal Setting

This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulation"

Differentiable signal processing on the sphere for PyTorch

Source code for paper "Learning spatiotemporal dynamics with a pretrained generative model" in Nature Machine Intelligence