0% completed

All LessonsFree Lessons (6)

Introduction

Course Introduction JAX Overview

JAX Programming Model

Pure Functions JAX and NumPy Just-in-Time (JIT) Compilation JAX Expressions (Jaxpr)Asynchronous Dispatch Auto-differentiation Pytrees Quiz: Basics of JAX Challenge: Basics of JAX

Linear Algebra

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

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 - Vectors and Matrices

Quiz - Vectors and Matrices

Let's test your knowledge of vectors and matrices in JAX with a quiz.

We'll cover the following...

Vectors and Matrices

1.

Which of the following options is not valid syntax for computing the inner product?

A.

c=a.b

B.

c=a@b

C.

c=jnp.inner(a,b)

D.

c=jnp.dot(a,b)

1 / 4