JAX Expressions (Jaxpr)

This lesson will provide an introduction to JAX Expressions.

We'll cover the following

We just discussed how trace objects are used to preempt the sequence of operations outlined in the function. These extracted JAX expressions are discussed in this lesson.

Lambda expressions

Before beginning the JAX expressions, it is helpful to have a quick introduction of Lambda expressions.

Lambda expressions are usually used for anonymous functions. Defining them is pretty easy:

  1. Specify the lambda keyword
  2. Outline the parameters (enclosed by :).
  3. Specify the expression body.

For example, a simple Lambda expression to calculate the circumference of a circle is:

Get hands-on with 1000+ tech skills courses.