• 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
  • »
  • Overview: module code

All modules for which code is available

  • functorch._src.eager_transforms
  • functorch._src.make_functional
  • functorch._src.vmap

© Copyright functorch Contributors.

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