Skip to content

functorch 0.2.1
Compare
Choose a tag to compare
@zou3519 zou3519 released this 05 Aug 20:26
v0.2.1
a6e0e61

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

Changelog

  • Previously the functorch package was incompatible with the PyTorch 1.12.0 cu102 package. This is now fixed (functorch 0.2.1 is compatible with all PyTorch 1.12.1 packages).
  • Fixed a 25% regression from v0.1.1 to v0.2.0 in computing hessians of fully-connected layers (#989)
  • Added batching rules for masked_fill (#946), searchsorted (#966)
  • Batch norm now works with all forms of vmap when training is False (or .eval() is set on the model) (#958)
Assets 14
Loading