Parallel Reductions with Zygote

We recently had some success using Folds here. Don’t know if this helps with Zygote though.