8
Diffusion for the curious
Turbocharge ML with JAX and TPUs
Welcome
1
TPUs
2
Introduction to JAX
3
Linear Regression in JAX
4
Exercise 1: MNIST in JAX
5
Flax Foundations
6
Exercise 2: Linear Regression in Flax
7
Let’s build a ResNet!
8
Diffusion for the curious
9
Stable Diffusion in JAX / Flax !
10
Vision Transformer
11
Exercise 1 Solution
12
Exercise 2 Solution
13
Resources
References
8
Diffusion for the curious
TODO: Add speaker notes
8.0.0.1
How do we learn the probabilities of what an image represents?
8.0.0.2
Adding and detecting noise
8.0.0.3
Enter the Unet
8.0.0.4
Embeddings
8.0.0.5
Latents
8.0.0.6
Multi-modal
8.0.0.7
Altogether
7
Let’s build a ResNet!
9
Stable Diffusion in JAX / Flax !