EnzymeAD/Enzyme.jl

API for Hessian

Open

#1.495 geöffnet am 6. Juni 2024

Auf GitHub ansehen
 (1 Kommentar) (1 Reaktion) (0 zugewiesene Personen)Julia (96 Forks)batch import
good first issue

Repository-Metriken

Stars
 (563 Stars)
PR-Merge-Metriken
 (Durchschn. Merge 5T 1h) (76 gemergte PRs in 30 T)

Beschreibung

A hessian / hessian! function in Enzyme would be cool!

Right now the method described in https://enzymead.github.io/Enzyme.jl/stable/generated/autodiff/#Forward-over-reverse uses BatchDuplicated. Am I correct in assuming that, like ForwardDiff's "chunks", Enzyme's "batches" are limited in size by memory bandwitch, typically to the order of 10 elements? If so, should there be a Hessian mode that computes HVPs in batches smaller than the size n?

Contributor Guide