Forward- and reverse-mode autodiff in JAX

Forward- and reverse-mode autodiff in JAX. JAX documentation.

Type

Web resource

Intellectual base

Web

Link

https://docs.jax.dev/en/latest/jacobian-vector-products.html

Cited by

1 citation across 1 artifact.

Domain-specific

• Automatic Differentiation
  • Forward mode is naturally efficient when few input directions are needed; reverse mode is naturally efficient when few output directions are needed, especially the gradient of a scalar objective with many parameters.

Verification

This reference passed the adversarial substantiation pipeline: it was checked to exist and to support the claim it is attached to. See how references were verified.

Registry ID ref:f36b1ea5b119 · see in the full table