Mini Map
Log In
Deep Learning with JAX and Flax
0%
1.
Course Introduction
About the Course
2.
Basics of JAX
Introduction to JAX Data Types and Array
Pure Functions and Random Numbers
Array Operations
Just-in-Time Compilation
Machine Learning with JAX
Challenge: Basics of JAX
Solution: Basics of JAX
Summary: Basics of JAX
Quiz: Basics of JAX
3.
Optimizers in JAX and Flax
Introduction to Optimizers
Adaptive Optimizers
Stochastic Gradient Descent
Challenge: Optimizers
Solution: Optimizers
Summary: Optimizers in JAX and Flax
Quiz: Optimizers
4.
Loss and Activation Functions
Introduction to Loss Functions
Types of Loss Functions in JAX
Computing and Monitoring Loss in JAX
Activation Functions in JAX
Challenge: Loss and Activation function
Solution: Loss and Activation function
Summary: Loss and Activation Functions
Quiz: Loss and Activation Functions
5.
Load Datasets in JAX
Preparing Text Dataset
Text Preprocessing and Sentiment Analysis
Loading Image Dataset
Challenge: Train and Test a Model
Solution: Train and Test a Model
Summary: Load Datasets in JAX
Quiz: Loading Datasets in JAX
6.
Image Classification and Distributed Training
Image Classification with JAX and Flax
Distributed Training with JAX and Flax
Challenge: Distributed Training with JAX and Flax
Solution: Distributed Training with JAX and Flax
Summary: Image Classification and Distributed Training
Quiz: Distributed Training with JAX and Flax
7.
TensorBoard and State Handling
Setting Up TensorBoard
How to Use TensorBoard with Flax
Track Model Training in JAX Using TensorBoard
Applying BatchNorm and DropOut Layers in JAX and Flax
Summary: TensorBoard and State Handling
Quiz: TensorBoard and State Handling
8.
LSTM in JAX and Flax
Data Preprocessing
LSTM Model
Challenge: LSTM in JAX and Flax
Solution: LSTM in JAX and Flax
Summary: LSTM in JAX and Flax
Quiz: LSTM in JAX and Flax
9.
Flax vs. TensorFlow
Major Functions in Flax and TensorFlow
Training model in Flax vs. TensorFlow
Summary: Flax vs. TensorFlow
Quiz: Flax vs. TensorFlow
10.
Using ResNet Model in Flax
Training the ResNet Model
ResNet Model Definition
Transfer Learning using ResNet Model
Summary: Using ResNet Model in Flax
Quiz: Using ResNet Model in Flax
Mini Project
Premium
Transfer Learning in JAX and Flax
11.
Conclusion
Wrap Up
12.
Appendix
Installing Packages
Sharing TensorBoard
Deep Learning with JAX and Flax
/
...
/
Summary: Flax vs. TensorFlow
Summary: Flax vs. TensorFlow
Let’s summarize the concepts we have learned so far.
We'll cover the following...
Recap
In this
...