JuliaDiff / JuliaDiff/ChainRules.jl
Rules for `kron`
Open
Nobody has claimed this yet.
good first issue
missing rule
- Dominant language
- Julia
- Stars
- 475
- Forks
- 98
- PR merge metrics
- No merged PRs in 30d
Description
Zygote currently punts this to AD, but that trick no longer works on nightly because kron(::AbstractVector, ::AbstractVector) stopped calling kron(::AbstractMatrix, ::AbstractMatrix).
Contributor guide
No contributing guide indexed for this repository
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 with the linked Zygote implementation in src/lib/array.jl and inspect how kron(::AbstractVector, ::AbstractVector) differs from the matrix case on nightly. Determine the ChainRules behavior needed for vector kron, then verify that the corresponding differentiation rules cover this case without relying on the old dispatch path.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- julia
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100