Module transforms

Implementations of function transformations.

Modules

• compile
  Compilation of functions.

Traits

• IntoGrad
  Trait for functions/closures that can be converted into a closure that computes the gradient.
• IntoKeyedValueAndGrad
  Similar to [IntoValueAndGrad] but for functions that take a hashmap of parameters.
• IntoValueAndGrad
  Trait for functions/closures that can be converted into a closure that computes the value and gradient.

Functions

• async_eval
  Asynchronously evaluate an iterator of Arrays.
• async_eval_params
  Asynchronously evaluate a module’s parameters.
• eval
  Evaluate an iterator of Arrays.
• eval_params
  Evaluate a module’s parameters.
• fallible_jvp
  Similar to jvp but handles closures that can return an error.
• fallible_vjp
  Similar to vjp but handles closures that can return an error.
• grad
  Returns a function which computes the gradient of f.
• jvp
  Compute the Jacobian-vector product.
• keyed_value_and_grad
  Returns a function which computes the value and gradient of f with keyed parameters.
• value_and_grad
  Returns a function which computes the value and gradient of f.
• vjp
  Compute the vector-Jacobian product.

Type Aliases

• KeyedGrad
  Type alias for a hashmap of gradients.
• KeyedParameters
  Type alias for a hashmap of parameters.