Skip to main content

One doc tagged with "jax"

View all tags

Automatic Differentiation

Forward- and reverse-mode automatic differentiation through computational graphs, illustrated with JAX transformations.