Big step forward: Zygote is now compatible with parallelism.
https://github.com/FluxML/Zygote.jl/pull/728
I won’t update here anymore, but you can track the issue here: https://github.com/SciML/DiffEqFlux.jl/issues/279
1 Like