Autograd in Flux

Hello!

It’s difficult to diagnose the issue here without a MWE, can you read through the instructions in Please read: make it easier to help you first?