• Install
  • functorch API Reference
    • Function Transforms
      • functorch.grad
      • functorch.grad_and_value
      • functorch.jacrev
      • functorch.vmap
      • functorch.vjp
    • Utilities for working with torch.nn.Modules
      • functorch.make_functional
      • functorch.make_functional_with_buffers
      • functorch.combine_state_for_ensemble
functorch
  • »
  • Search

© Copyright functorch Contributors.

Built with Sphinx using a theme provided by Read the Docs.