enzyme_sparse_accumulate and co in Julia

Repository-Metriken

Stars

563 Sterne (563 Sterne)

PR-Merge-Metriken

Durchschn. Merge 5T 1h (Durchschn. Merge 5T 1h)76 gemergte PRs in 30 T (76 gemergte PRs in 30 T)

Beschreibung

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

Contributor Guide

Research-Richtung

Das Issue fordert die Implementierung von Unterstützung für dünnbesetzte Jacobi Matrizen in Enzyme.jl. Derzeit gibt die jacobian Funktion dichte Matrizen zurück, selbst für dünnbesetzte Eingaben. Um fortzufahren, sollte der Quellcode von Enzyme.jl, insbesondere die Jacobi Berechnungsfunktionen, untersucht werden, um zu identifizieren, wo die Behandlung dünnbesetzter Matrizen integriert werden könnte. Zudem sollte erforscht werden, wie SparseArrays.jl mit Enzyme interagiert, und die Maintainer für Designrichtlinien konsultiert werden.

Tech Stack

Keine

Domain

backend

Issue Type

Funktion

SchwierigkeitGeschätzte Implementierungsschwierigkeit für neue Contributors, von 1 für sehr kleine Änderungen bis 5 für Expertenarbeit.

4

Geschätzte ZeitEin grober Zeitrahmen, um zu recherchieren, zu implementieren, zu testen und einen Pull Request vorzubereiten.

1-2 Tage

AktivitätsstatusWie verfügbar das Issue gerade wirkt: frisch, aktiv, stale, blockiert oder wartet auf Maintainer-Input.

Frisch

KlarheitWie klar das Issue die erwartete Änderung, Akzeptanzkriterien und den nächsten Schritt erklärt.

Recherche nötig

Voraussetzungen

JuliaEnzymeSparseArraysAutomatic differentiation

EinsteigerfreundlichkeitEin Score von 1 bis 100, der schätzt, wie zugänglich dieses Issue für First-Time Contributors ist.

25