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

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

Quiz: PRNG and Probability Distributions

Let's test our understanding of PRNG and probability distributions in JAX.

1

Which of the following statements is true about CDF, $F_X(x)$ ?

A)

$F_X(x)$ is left-continuous.

B)

$F_X(x)$ is non-decreasing.

C)

$F_X(x)$ can take any value from $-\infty$ to $\infty$ .

D)

It cannot be used with discrete distributions.

Question 1 of 40 attempted

Get hands-on with 1400+ tech skills courses.