JuliaDiff / JuliaDiff/DifferentiationInterface.jl
Pullback closure for `DifferentiateWith`
Nobody has claimed this yet.
- Dominant language
- Julia
- Stars
- 313
- Forks
- 35
- PR merge metrics
- No merged PRs in 30d
Description
At the moment, DI only has value_and_pullback, which computes the primal and the cotangent at once. Most rule definition tools require the pullback to be split into a closure so that it can be applied at a later point in time. This would give a roughly x2 performance increase for DifferentiateWith.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by tracing the existing value_and_pullback path used by DifferentiateWith and identify where the primal and cotangent are computed together. Examine how the pullback could be retained for later application, then verify that the deferred operation preserves existing behavior and measure whether the reported performance improvement is achieved.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- julia
- Domain
- backend-api-design
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 30/100