Machine Learning with Flax - From Zero to Hero

Machine Learning with Flax - From Zero to Hero

Aleksa Gordić - The AI Epiphany via YouTube Direct link

Intro - Flax is performant and reproducible

1 of 15

1 of 15

Intro - Flax is performant and reproducible

Class Central Classrooms beta

YouTube playlists 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. 1 Intro - Flax is performant and reproducible
  2. 2 Deepnote walk-through sponsored
  3. 3 Flax basics
  4. 4 Flax vs Haiku
  5. 5 Benchmarking Flax
  6. 6 Linear regression toy example
  7. 7 Introducing Optax Adam state example
  8. 8 Creating custom models
  9. 9 self.param example
  10. 10 self.variable example
  11. 11 Handling dropout, BatchNorm, etc.
  12. 12 CNN on MNIST example
  13. 13 TrainState source code
  14. 14 CNN dropout modification
  15. 15 Outro and summary

Never Stop Learning.

Get personalized course recommendations, track subjects and courses with reminders, and more.

Someone learning on their laptop while sitting on the floor.