Shortcuts

Source code for torch.func

from torch._functorch.eager_transforms import (
    grad_and_value,
    vjp,
    jvp,
    jacrev,
    jacfwd,
    hessian,
    functionalize,
    linearize
)
from torch._functorch.apis import grad
from torch._functorch.functional_call import functional_call, stack_module_state
from torch._functorch.batch_norm_replacement import replace_all_batch_norm_modules_
from torch._functorch.apis import vmap

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources