What's Changed
- v0.6: drop support for <2.0, support jacobian and hessian computation by @yoyololicon in #12
Summary:
- Drop support for
torch<2.0
to use the new schema for writingautograd.Function
- add
vmap
for calculatingjacfwd/jacrev/hessian
Full Changelog: v0.5...v0.6