AI Features
Log In
0% completed
All Lessons
Free Lessons (3)
Course Introduction
About the Course
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
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
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
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
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
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
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
Flax vs. TensorFlow
Major Functions in Flax and TensorFlow
Training model in Flax vs. TensorFlow
Summary: Flax vs. TensorFlow
Quiz: Flax vs. TensorFlow
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
Transfer Learning in JAX and Flax
Conclusion
Wrap Up
Appendix
Installing Packages
Sharing TensorBoard
Home
Courses
Deep Learning with JAX and Flax
Challenge: Optimizers
Test your understanding by training a model using multiple optimizers.
We'll cover the following...
Problem statement
Try it yourself
Problem statement
...
Ask