Official implementation of paper MiAD: Mirage Atom Diffusion for De Novo Crystal Generation

Official Jax Implementation of MaskGIT

Environments and Algorithms for Generative Flow Networks in JAX

Respository for the paper "Matrix-Free Two-to-Infinity and One-to-Two Norms Estimation" (AAAI-2026)

TorchCFM: a Conditional Flow Matching library

Learning from synthetic data - code and models

An open source implementation of CLIP.

An interactive introduction to GFlowNets

Official implementation of Progressive Tempering Sampler with Diffusion (ICML 2025)

A GFlowNet with a chemical synthesis action space.

Elucidating the Design Space of Diffusion-Based Generative Models (EDM)

Code for MGDM algorithm, ICML 2025, https://arxiv.org/abs/2502.03332

Generative Flow Networks - GFlowNet

code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"

[ICLR 2022] Path integral sampler

Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)

[ICLR 2025] Official implementation for "Underdamped Diffusion Bridges with Applications to Sampling".

A curated list of resources about generative flow networks (GFlowNets).

A collection of resources and papers on diffusion samplers.

Official implementation of paper: SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-training

The official implementation of "Smoothie: Smoothing Diffusion on Token Embeddings for Text Generation"

AlphaFold 3 inference pipeline.