A function transformation is an operation that takes a function as input and returns a modified function as output. In JAX and related systems, function transformations support capabilities such as automatic differentiation, vectorization, compilation, and parallelization.