Back To Course Home

Introduction to JAX and Deep Learning

0% completed

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

Quiz: Convolution

Let's test out knowledge of convolutions and transposed convolutions by completing this quiz.

Feel free to use the code playground provided below to verify your answers.

1

Suppose a filter of $3\times3$ is applied on a $64\times64$ image. What is the output size if we have a padding of 2 (in each direction)?

A)

$60\times60$

B)

$61\times61$

C)

$64\times64$

D)

$63\times63$

Question 1 of 20 attempted

Get hands-on with 1400+ tech skills courses.