functorch 0.1.1

We’re excited to present the functorch 0.1.1 minor bug-fix release, compatible with PyTorch 1.11. Please see here for installation instructions.

Changelog

Fixed a bug when composing jvp with vmap (#603)
jvp now works when called inside autograd.Function (#607)
make_functional (and variants) now work with models that do parameter sharing (also known as weight tying) (#620)
Added batching rules for nn.functional.silu, nn.functional.prelu, nn.functional.glu (#677, #609, #665)
Fixed vmap support for nn.functional.group_norm, binomial, torch.multinomial, Tensor.to (#685, #670, #672, #649)