LuxDL/Lux.jl

Enzyme wrappers for AD utilities in Lux

Aperta

#738 aperta il 27 giu 2024

Vedi su GitHub
 (0 commenti) (1 reazione) (0 assegnatari)Julia (84 fork)batch import
enzymegood first issue

Metriche repository

Star
 (700 stelle)
Metriche merge PR
 (Merge medio 13g 21h) (2 PR mergiate in 30 g)

Descrizione

Functions

  • batched_jacobian
    • Enzyme.Forward
    • Enzyme.Reverse
  • vector_jacobian_product
  • jacobian_vector_product

Nested Autodiff

Enzyme needs us to use autodiff_thunk for the internal call, so we can easily replace the call inside custom rrules.

  • EnzymeRules
  • CRC.rrule

Guida contributor

Direzione di ricerca
La richiesta riguarda l'implementazione di wrapper Enzyme per diverse utility di differenziazione automatica, tra cui batched jacobian, vector jacobian product, jacobian vector product, e la gestione della differenziazione automatica annidata tramite autodiff thunk. Un contributore dovrebbe studiare il pacchetto Enzyme.jl esistente e l'interfaccia AD di Lux.jl, quindi implementare ciascuna funzione seguendo gli schemi di librerie simili. Non sono presenti PR o commenti collegati; il lavoro iniziale può iniziare forkando il repository e facendo riferimento all'issue.
Tech stack
Nessuno
Dominio
machine learningai
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
Mezza giornata
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Nuova
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Abbastanza chiara
Prerequisiti
JuliaAutomatic differentiationEnzyme
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
30