Multimodal Image Generator

Train a VQ-VAE image tokenizer and an autoregressive text-conditioned transformer with guided sampling.

Hard5.3 hours64 steps

Overview

Train a VQ-VAE image tokenizer and an autoregressive text-conditioned transformer with guided sampling.

What you'll learn

Generative modeling JAX Multimodal learning

FULL WALKTHROUGH

Learn every step on its own page

This project is no longer compressed into a few chapters. Open the dedicated learning workspace for a lesson-by-lesson explanation with concepts, MathJax mathematics, code, tests, mistakes, checkpoints, and persistent navigation.

Open 64-step walkthrough →

Build progress

0 / 64 steps0%

Move the tracker as you finish the original Deep-ML steps. Reaching 100% unlocks the completion action and certificate.

Architecture

Work through the system one dependable layer at a time. Each stage feeds the next and remains independently testable.

01VQ-VAE→

02Discrete tokens→

03Classifier-free guidance

∑

Mathematics & visual explanation

Translate the core equations into code, validate intermediate tensors, and compare the implementation with a small numerical reference.

objective(θ) = data_term(θ) + λ · regularization(θ)

The exact objective evolves with each milestone. Keep a notebook of shapes, invariants, and numerical checks.