Skip to content

DifferentiationInterface-v0.5.6

Compare
Choose a tag to compare
@gdalle gdalle released this 27 Jun 05:46
· 189 commits to main since this release
ff529cb

TLDR

A lot of internal changes to allow for batched mode pushforward, pullback and hvp (propagating several seeds at once) in #320 and other PRs. The end result is that jacobian and hessian should be faster for the backends that support it, in both dense and sparse settings.

At the moment, batched mode is not public API, and only works with ForwardDiff (by storing several Partials in a Dual number), but I plan to add Enzyme soon (leveraging BatchDuplicated). When I say "not public API", I mean that users will reap the benefits by default when they call jacobian or hessian, but shouldn't rely on the lower-level primitives pushforward, pullback and hvp to work in batched mode (this behavior is not documented anyway).

What's Changed

Full Changelog: DifferentiationInterface-v0.5.5...DifferentiationInterface-v0.5.6