enzyme_sparse_accumulate and co in Julia

Métriques du dépôt

Stars

563 étoiles (563 étoiles)

Métriques de merge PR

Merge moyen 5j 1h (Merge moyen 5j 1h)76 PRs mergées en 30 j (76 PRs mergées en 30 j)

Description

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

Guide contributeur

Direction de recherche

La demande concerne l'implémentation de la prise en charge des matrices jacobiennes creuses dans Enzyme.jl. Actuellement, la fonction jacobian renvoie des matrices denses même pour des entrées creuses. Pour avancer, il faut examiner le code source d'Enzyme.jl, en particulier les fonctions de calcul de la jacobienne, et identifier où le traitement des matrices creuses pourrait être intégré. De plus, étudier comment SparseArrays.jl interagit avec Enzyme et consulter les mainteneurs pour des conseils de conception.

Stack technique

Aucun

Domaine

backend

Type d'issue

Fonctionnalité

DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.

4

Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.

1-2 jours

Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.

Récente

ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.

Investigation requise

Prérequis

JuliaEnzymeSparseArraysAutomatic differentiation

Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.

25