EnzymeAD/Enzyme.jl

API for Hessian

Open

#1.495 aberto em 6 de jun. de 2024

Ver no GitHub
 (1 comment) (1 reaction) (0 assignees)Julia (96 forks)batch import
good first issue

Métricas do repositório

Stars
 (563 stars)
Métricas de merge de PR
 (Mesclagem média 5d 1h) (76 fundiu PRs em 30d)

Description

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?

Guia do colaborador

Direção de pesquisa
Examine a viabilidade de implementar uma função Hessiana em Enzyme.jl que calcule produtos Hessian vetor em lotes para lidar com Hessianas grandes, com base no modo forward over reverse existente.
Pilha de tecnologia
None
Domain
machine learning
Tipo Issue
Funcionalidade
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-2 dias
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
JuliaAutomatic Differentiation
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
50