LuxDL/Lux.jl

Enzyme wrappers for AD utilities in Lux

開放

#738 建立於 2024年6月27日

在 GitHub 查看
 (0 則留言) (1 個反應) (0 位負責人)Julia (84 個分叉)batch import
enzymegood first issue

倉庫指標

星標
 (700 顆星)
PR 合併指標
 (平均合併 13天 21小時) (30 天內合併 2 個 PR)

描述

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

貢獻者指南

研究方向
該問題要求為多個自動微分工具實作 Enzyme 包裝器，包括 batched jacobian、vector jacobian product、jacobian vector product，並透過 autodiff thunk 處理巢狀自動微分。貢獻者應研究現有的 Enzyme.jl 套件和 Lux.jl 的自動微分介面，然後參照類似函式庫的模式實作每個函式。沒有關聯的 PR 或評論；初始工作可以從 fork 儲存庫並引用該問題開始。
技術棧
無
領域
machine learningai
議題類型
功能
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
半天
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
新近可參與
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
大致清晰
前置要求
JuliaAutomatic differentiationEnzyme
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
30