Year-End Discount: 10% OFF 1-year and 20% OFF 2-year subscriptions!

Introduction to JAX and Deep Learning

1 Project
53 Lessons
7 Quizzes
14 Challenges
95 Playgrounds
20 Illustrations

Takeaway Skills

Learn the basics of JAX

Learn how to apply Autograd

Use auto vectorization for batching

Use Haiku and Flax for implementing neural networks

Cover Optax and overview of common optimization algorithms in deep learning

Use Chex for testing JAX programs

Learn the basics of applied linear algebra

Learn random variables theory and probability distributions

Learn pseudo-random number generation

Cover the basics of optimal transport

Learn the basics of JAX

Learn how to apply Autograd

Use auto vectorization for batching

Use Haiku and Flax for implementing neural networks

Cover Optax and overview of common optimization algorithms in deep learning

Use Chex for testing JAX programs

Learn the basics of applied linear algebra

Learn random variables theory and probability distributions

Learn pseudo-random number generation

Cover the basics of optimal transport

Course Overview

JAX is a Python library designed for high-performance ML research. It is a powerful numerical computing library, just like Numpy, but with some key improvements. In this course, you will learn all about JAX and its ecosystem of libraries (Haiku, Jraph, Chex, Flax, Optax). Addressing a...

How You'll Learn

Hands-on coding environments

You don’t get better at swimming by watching others. Coding is no different. Practice as you learn with live code environments inside your browser.

Faster than videos

Videos are holding you back. The average video tutorial is spoken at 150 words per minute, while you can read at 250. That‘s why our courses are text-based.

No set-up required

Start learning immediately instead of fiddling with SDKs and IDEs. It‘s all on the cloud.

Progress you can show

Built in assessments let you test your skills. Completion certificates let you show them off.

Hands-on coding environments

You don’t get better at swimming by watching others. Coding is no different. Practice as you learn with live code environments inside your browser.

Faster than videos

Videos are holding you back. The average video tutorial is spoken at 150 words per minute, while you can read at 250. That‘s why our courses are text-based.

No set-up required

Start learning immediately instead of fiddling with SDKs and IDEs. It‘s all on the cloud.

Progress you can show

Built in assessments let you test your skills. Completion certificates let you show them off.

Join more than 1.6 million learners from companies like

What our learners are saying

What our learners are saying

COURSE BY:

Educative

Teams of every size choose
Educative for Business
to train, upskill & onboard their developers