Is this impossible for Zygote Adjoint to handle

Sounds like it’s missing some kind of reduction somewhere. It’s expecting a float instead it got an array

We’ll need something more minimal to give you a straightforward answer. I’m my experience most times it’s something in the user script that needs to be changed.