EnzymeAD/Enzyme.jl

enzyme_sparse_accumulate and co in Julia

Aperta

#1947 aperta il 9 ott 2024

Vedi su GitHubGuida contributor
 (0 commenti) (0 reazioni) (0 assegnatari)Julia (96 fork)batch import
help wanted

Metriche repository

Star
 (563 stelle)
Metriche merge PR
 (Merge medio 5g 1h) (76 PR mergiate in 30 g)

Descrizione

What would we need to do to define support for sparse operations/jacobians in Enzyme.jl?

using Enzyme
using SparseArrays

julia> Enzyme.jacobian(Enzyme.Forward, x->sum(abs2, x, dims=2), sprandn(12, 0.5))
12×12 Matrix{Float64}:
  0.0       0.0       0.0      0.0     0.0     0.0     0.0     0.0      0.0  0.0  0.0  0.0
 -1.88261  -1.88261   0.0      0.0     0.0     0.0     0.0     0.0      0.0  0.0  0.0  0.0
  0.0       0.0      -3.04331  0.0     0.0     0.0     0.0     0.0      0.0  0.0  0.0  0.0
  0.0       0.0       0.0      0.0     0.0     0.0     0.0     0.0      0.0  0.0  0.0  0.0
  0.0       0.0       0.0      0.0     0.0     0.0     0.0     0.0      0.0  0.0  0.0  0.0
  0.0       0.0       0.0      0.0     0.0     0.0     0.0     0.0      0.0  0.0  0.0  0.0
  0.0       0.0       0.0      1.2327  1.2327  1.2327  1.2327  0.0      0.0  0.0  0.0  0.0
  0.0       0.0       0.0      0.0     0.0     0.0     0.0     1.63961  0.0  0.0  0.0  0.0
  0.0       0.0       0.0      0.0     0.0     0.0     0.0     0.0      0.0  0.0  0.0  0.0
  0.0       0.0       0.0      0.0     0.0     0.0     0.0     0.0      0.0  0.0  0.0  0.0
  0.0       0.0       0.0      0.0     0.0     0.0     0.0     0.0      0.0  0.0  0.0  0.0
  0.0       0.0       0.0      0.0     0.0     0.0     0.0     0.0      0.0  0.0  0.0  0.0

Guida contributor

Direzione di ricerca
La richiesta riguarda l'implementazione del supporto per Jacobiane sparse in Enzyme.jl. Attualmente, la funzione jacobian restituisce matrici dense anche per input sparsi. Per procedere, si dovrebbe esaminare il codice sorgente di Enzyme.jl, in particolare le funzioni di calcolo della Jacobiana, e identificare dove potrebbe essere integrata la gestione sparsa. Inoltre, studiare come SparseArrays.jl interagisce con Enzyme e consultare i maintainer per indicazioni di progettazione.
Tech stack
Nessuno
Dominio
backend
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
4
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-2 giorni
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Nuova
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Richiede indagine
Prerequisiti
JuliaEnzymeSparseArraysAutomatic differentiation
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
25