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: Convolution

Quiz: Convolution

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

We'll cover the following...

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