JAX and NumPy

This lesson will explore the relationship between JAX and NumPy, since JAX can be treated as a parallelized version of NumPy.

This lesson assumes a certain level of familiarity with NumPy. Let’s first start with how normal NumPy and JAX relate.

JAX Numpy

JAX has its own variant of NumPy, which we can import as:

Get hands-on with 1200+ tech skills courses.