framework.
Read More


torch.func Whirlwind Tour


What is torch.func?


Why composable function transforms?


What are the transforms?


torch.func API Reference


Function Transforms


Utilities for working with torch.nn.Modules


UX Limitations


General limitations


torch.autograd APIs


vmap limitations


Randomness


Migrating from functorch to torch.func


function transforms


NN module utilities


functorch.compile

