jax-js
    Preparing search index...

    Function vjp

    • Calculate the reverse-mode vector-Jacobian product for a function.

      Type Parameters

      Parameters

      Returns [
          ReturnType<F>,
          (
              cotangents: MappedJsTree<ReturnType<F>, Array, ArrayLike>,
          ) => MappedJsTree<Parameters<F>, ArrayLike, Array>,
      ]