-
Flux has a lot of gradient definitions that we need to port over to Zygote. Copying them over would be an easy patch for anyone interested in contributing.
-
In the docs of [AbstractFFTs.jl](https://juliamath.github.io/AbstractFFTs.jl/stable/api/#AbstractFFTs.plan_fft), it says that it is preferrable to apply the FFT plans to pre-allocated arrays using `mu…
-
Bug is:
```
$ futhark dev --gpu step.fut > step.ir -v
[ +0.000037] Reading and type-checking source program
[ +0.155007] Defunctorising
[ +0.000759] Monomorphising
[ +0.024246] Lifting lam…
-
This is a performance bug, because the code version that uses these arrays may never be executed.
I assume this happens also for clean-ad since it has nothing to do with matrix multiplication.
To …
-
Enzyme's integration into Rust should give the user
- precise control of what are constant, primary and adjoint variables
- make it easy to mark a function as a candidate for differentiation
- i…
-
What is the recommended way here to use a custom adjoint for an objective and/or constraints defined using ChainRulesCore.jl for the use in the value_jacobian function? Lets say I have defined my adjo…
-
https://youtu.be/KCTfPyVIxpc?t=2730
https://github.com/mitmath/18337/blob/6b9e8d02b44ec02554be7861c6c3381e5f2de726/lecture11/adjoints.jmd#L273-L276
-
We should probably redesign differentiation in a more organised way, likely following the `ChainRules.jl` ideas or even using `ChainRules.jl` directly. In particular this should enable us to leverage …
-
## Description
(This issue is a revival of #1257. Essentially all of what follows is cribbed from @charlesm93!)
It may be possible to improve how derivatives are propagated through the the algebra…
jgaeb updated
3 years ago
-
I am struggling with defining custom `rrule` and computing gradients a custom `AbstractMatrix` subtype.
For simplicity, assume naive matmul wrapped in a struct
```julia
using Flux, Zygote, Chai…