Which direction: DifferentiatonInterface, Enzyme, Zygote with CUDA and FFTs?

A more detailed autodiff tutorial in Reactant Automatic Differentiation | Reactant.jl

4 Likes