Can we use DualArrays.jl to compute gradients of neural networks as defined in Lux?
Can we use DualArrays.jl to compute gradients of neural networks as defined in Lux?