Skip to content
This repository was archived by the owner on Aug 21, 2025. It is now read-only.

functorch 0.1.1

Choose a tag to compare

View all tags
@zou3519 zou3519 released this 12 Apr 23:32
v0.1.1
92f1610

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)
Assets 10
Loading