Completed
Introducing Optax Adam state example
Class Central Classrooms beta
YouTube videos curated by Class Central.
Classroom Contents
Machine Learning with Flax - From Zero to Hero
Automatically move to the next video in the Classroom when playback concludes
- 1 Intro - Flax is performant and reproducible
- 2 Deepnote walk-through sponsored
- 3 Flax basics
- 4 Flax vs Haiku
- 5 Benchmarking Flax
- 6 Linear regression toy example
- 7 Introducing Optax Adam state example
- 8 Creating custom models
- 9 self.param example
- 10 self.variable example
- 11 Handling dropout, BatchNorm, etc.
- 12 CNN on MNIST example
- 13 TrainState source code
- 14 CNN dropout modification
- 15 Outro and summary
