JAX Expressions (Jaxpr)
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.
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:
- Specify the
- Outline the parameters (enclosed by
- Specify the expression body.
For example, a simple Lambda expression to calculate the circumference of a circle is: