Open st-- opened 2 years ago
The issue seems to be that Zygote does not properly differentiate through foldl()
[returning nothing
] :( no idea why it's not working, there's implementations such as https://github.com/JuliaDiff/ChainRules.jl/pull/526 ...
😞
Currently our build is broken due to a Zygote AD issue. To unblock other PRs, let's uncomment the test (see #414) and investigate the issue separately.