That was a recently introduced Zygote.jl bug from a change in accumulation. I believe @ToucheSir would have more details. I think v0.6.42 would be a working version?
We’re working through this right now.
1 Like