Search — functorch 0.1.1 documentation

Tutorials
Examples
GitHub

Table of Contents

0.1.1 (works with PyTorch 1.11) ▼

Getting Started

Install functorch
Whirlwind Tour
UX Limitations

API Reference and Notes

functorch
functorch.compile (experimental)

Tutorials

Jacobians, Hessians, hvp, vhp, and more: composing functorch transforms
Model ensembling
Per-sample-gradients
Neural Tangent Kernels
AOT Autograd - How to use and optimize?

Docs >
Search

Shortcuts

© Copyright functorch Contributors.

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