0% completed

All LessonsFree Lessons (6)

Introduction

Course Introduction JAX Overview

JAX Programming Model

Just-in-Time (JIT) Compilation

JAX Expressions (Jaxpr)

Asynchronous Dispatch

Auto-differentiation

Quiz: Basics of JAX

Challenge: Basics of JAX

Linear Algebra

Vectors Matrices More on Matrices Quiz - Vectors and Matrices Challenge: Vectors and Matrices Auto Vectorization Auto Parallelization Multivariate Calculus Quiz: Multivariate Calculus, Auto Vectorization/Parallelization Challenge II: Linear Regression Convolutions Two-dimensional Convolutions Transposed Convolutions Quiz: Convolution Challenge: Convolution

Random Variables and Distributions

Pseudo-Random Number Generation Continuous Probability Distributions Discrete Distributions Quiz: PRNG and Probability Distributions Challenge: PRNG and Probability Distribution Divergence Measures Introduction to Optimal Transport and OTT

JAX Ecosystem

Introduction to JAX Ecosystem Introduction to Flax and Linen Flax: Linear Modules Flax: Overview of Convolution and Sequence Models Quiz: Flax Challenge: Flax Haiku Introduction to Optax and Optimizers Optax: Advanced Features Challenge: Optax Jraph Chex Quiz: The JAX Ecosystem Conclusion

Mini Project

Project: GAN Using the JAX ecosystem

Appendix

Installation Awesome JAX Libraries Awesome JAX-based Models Vector Calculus - II Common Tracer Errors Glossary

Introduction to JAX and Deep Learning/

...

/

Quiz: Basics of JAX

Quiz: Basics of JAX

Test your knowledge of the JAX programming model by completing this quiz.

We'll cover the following...

Technical Quiz

1.

Which of these libraries is not a part of the JAX ecosystem?

A.

Haiku

B.

Optax

C.

Jadex

D.

RLax

1 / 7