I think it doesn’t work with FFT : ForwardDiff and Zygote cannot automatically differentiate (AD) function from C^n to R that uses FFT
Thank you for the post. Since I am new to Zygote and other related packages I really don’t know what should I try first? For my case, do I need to define gradient rules manually ? If so, this will be out of my knowledge and I crash into a serious issue.
B.R.