All Examples¤

Quick Start¤

VAE on MNIST

Beginner

Train a Variational Autoencoder on MNIST digits

VAE Image MNIST

⏱ 10 min

GAN on MNIST

Beginner

Train a Generative Adversarial Network on MNIST

GAN DCGAN MNIST

⏱ 15 min

Diffusion on MNIST

Beginner

Train a DDPM diffusion model on MNIST digits

Diffusion DDPM MNIST

⏱ 15 min

Diffusion MNIST Training

Beginner

Complete training workflow for diffusion models on MNIST

Diffusion Training MNIST

⏱ 20 min

Flow on MNIST

Beginner

Train a normalizing flow model on MNIST

Flow RealNVP MNIST

⏱ 10 min

Diffusion Models¤

Simple Diffusion

Intermediate

Introduction to diffusion models with DDPM

DDPM Denoising Image

⏱ 20 min

DiT Demo

Advanced

Scalable diffusion with Diffusion Transformers

DiT Transformer Scalable

⏱ 30 min

Advanced Tutorials¤

Advanced Training Pipeline

Intermediate

Production-ready training with optimizers, schedulers, and checkpointing

Training Optimizer Scheduler Checkpointing

⏱ 2 min

Advanced VAE

Advanced

β-VAE, VQ-VAE, and disentanglement techniques

β-VAE VQ-VAE Disentanglement

⏱ 45 min

Advanced GAN

Advanced

WGAN, StyleGAN, and progressive training

WGAN StyleGAN Progressive

⏱ 45 min

Advanced Diffusion

Advanced

DDIM, latent diffusion, and guidance techniques

DDIM Latent Guidance

⏱ 50 min

Advanced Flow

Advanced

Neural spline flows, MAF, and IAF architectures

Spline MAF IAF

⏱ 40 min

Specialized Models¤

Energy-Based Model

Advanced

EBM training with MCMC and Langevin sampling

EBM MCMC Langevin

⏱ 25 min

Audio Generation

Intermediate

Generate audio with generative models

Audio WaveNet Spectrogram

⏱ 30 min

Simple Text Generation

Beginner

Character-level text generation with temperature sampling

Text Character-Level RNN Temperature

⏱ 5 min

Simple Image-Text Multimodal

Intermediate

Multimodal learning with image and text encoders

Multimodal Image Text Retrieval

⏱ 10 min

Geometric Benchmark

Advanced

Geometric generative model benchmarking

3D Point Cloud Mesh

⏱ 45 min

Geometric Models¤

Geometric Models Demo

Beginner

Quick reference for point clouds, meshes, and voxels

Point Cloud Mesh Voxel

⏱ 5 min

Geometric Losses Demo

Intermediate

Loss functions for point clouds, meshes, and voxels

Loss Functions Point Cloud Mesh Voxel

⏱ 15 min

Simple Point Cloud

Beginner

Generate and visualize 3D point clouds with transformers

Point Cloud Transformer 3D

⏱ 10 min

Geometric Benchmark

Advanced

Comprehensive evaluation on geometric tasks

Benchmark ShapeNet Metrics

⏱ 45 min

Protein Modeling¤

Protein Diffusion

Advanced

Generate 3D protein structures with diffusion models

Diffusion 3D Generation Constraints Visualization

⏱ 5 min

Protein Tech Validation

Beginner

Validate your environment for protein diffusion modeling

Validation Setup JAX

⏱ 5 min

Protein Extensions

Intermediate

Add domain-specific constraints with protein extensions

Extensions Constraints Bonds

⏱ 10 min

Protein Model Extension

Intermediate

Combine point cloud models with protein extensions

Protein Extensions Point Cloud

⏱ 10 min

Protein Model with Modality

Intermediate

Using the modality architecture for protein models

Protein Modality Factory

⏱ 10 min

Protein Point Cloud

Intermediate

Point cloud modeling with geometric constraints

Protein Point Cloud Constraints

⏱ 15 min

Protein Extensions

Intermediate

Using protein extensions with configuration system

Extensions Config Protein

⏱ 10 min

Protein-Ligand Benchmark

Advanced

Protein-ligand binding site generation

Protein SE(3) Equivariant

⏱ 60 min

Framework & Techniques¤

BlackJAX Integration

Intermediate

MCMC sampling with BlackJAX: HMC, NUTS, and MALA algorithms

MCMC HMC NUTS BlackJAX

⏱ 15 min

BlackJAX Sampling Examples

Advanced

Compare HMC, MALA, NUTS samplers and direct BlackJAX API usage

MCMC Comparison Performance

⏱ 20 min

BlackJAX Integration Examples

Advanced

Direct API vs functional API: progress bars, JIT compilation, and performance

Integration Performance JIT

⏱ 25 min

Loss Functions

Intermediate

Comprehensive guide to Workshop loss functions

Losses KL Adversarial

⏱ 20 min

Framework Features

Intermediate

Explore Workshop's architectural patterns

Architecture Patterns Framework

⏱ 30 min

β-VAE Benchmark

Advanced

Compare β-VAE configurations and disentanglement

β-VAE Benchmark Tuning

⏱ 40 min

Reference Tables¤

📊 Browse by Category

By Model Type¤

Model	Examples	Level
VAE	Basic, Advanced, β-VAE Benchmark	⭐ - ⭐⭐⭐
GAN	Basic, Advanced	⭐ - ⭐⭐⭐
Diffusion	Basic, Training, Simple, DiT, Advanced	⭐ - ⭐⭐⭐
Flow	Basic, Advanced	⭐ - ⭐⭐⭐
EBM	Simple EBM	⭐⭐⭐
Text	Simple Text Generation	⭐
Multimodal	Image-Text	⭐⭐
Protein	Modality, Point Cloud, Extensions, Ligand	⭐⭐ - ⭐⭐⭐
Geometric	Benchmark	⭐⭐⭐

By Dataset¤

Dataset	Examples
MNIST	VAE · GAN · Diffusion · Diffusion Training · Flow
Text	Simple Text Generation
Multimodal	Image-Text
Audio	Audio Generation
Protein/3D	Modality · Point Cloud · Extensions · Ligand · Geometric