Introduction to JAX and Deep Learning

53 Lessons
7 Quizzes
14 Challenges
95 Playgrounds
20 Illustrations
Get real world experience by completing the Mini Projects included in the course

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.3 million learners from companies like

COURSE BY:

Educative

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

Learn in-demand tech skills in half the time

Copyright ©2022 Educative, Inc. All rights reserved.

soc2