jax-js
    Preparing search index...

    Function linearize

    • Produce a local linear approximation to a function at a point using jvp() and partial evaluation.

      Type Parameters

      • F extends (...args: any[]) => JsTree<Array>
      • HA extends boolean = false

      Parameters

      Returns HA extends true
          ? ReturnType<F> extends [Out, Aux]
              ? [
                  Out,
                  OwnedFunction<
                      (...tangents: MappedJsTree<Parameters<F>, Array, ArrayLike>) => Out,
                  >,
                  Aux,
              ]
              : never
          : [
              ReturnType<F>,
              OwnedFunction<
                  (
                      ...tangents: MappedJsTree<Parameters<F>, Array, ArrayLike>,
                  ) => ReturnType<F>,
              >,
          ]