FluxML/Zygote.jl

Convert all adjoints to rrules

Open

#1,209 建立於 2022年4月20日

在 GitHub 查看
 (3 留言) (5 反應) (0 負責人)Julia (1,568 star) (220 fork)batch import
ChainRulesenhancementgood first issueup for grabs

描述

We have ZygoteRuleConfig for this, so there should be no major technical limitations. rrules generally have far better UX than @adjoints (as anyone who has had to read a stacktrace from Zygote can attest). The ultimate end goal would be to get rid of ZygoteRules entirely.

貢獻者指南

技術棧
無
領域
backendmachine learning
議題類型
refactor
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
5
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
over 1 week
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
stale
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
mostly clear
前置要求
Julia programmingZygote.jl internalsrrule and @adjoint concepts
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
10
研究方向
Identify all existing @adjoint definitions in the Zygote.jl codebase. For each one, write an equivalent rrule using ZygoteRuleConfig, ensuring correct behavior. Test with ChainRulesTestUtils and verify that stack traces improve. The conversion is systematic but requires deep understanding of both old and new APIs. No specific files are highlighted in the issue; maintainers have not provided further guidance beyond the title and description.