Shuffled gradient without prep

Author: gdalle

DifferentiationInterface/src/first_order/gradient.jl

@@ -133,6 +133,19 @@ function shuffled_gradient(
     return gradient(f, prep, backend, x, rewrap(unannotated_contexts...)...)
 end
 
+function shuffled_gradient!(
+    grad,
+    x,
+    f::F,
+    backend::AbstractADType,
+    rewrap::Rewrap{C},
+    unannotated_contexts::Vararg{Any,C},
+) where {F,C}
+    gradient!(f, grad, backend, x, rewrap(unannotated_contexts...)...)
+    return nothing
+end
+
+#=
 function shuffled_gradient!(
     grad,
     x,
@@ -145,3 +158,4 @@ function shuffled_gradient!(
     gradient!(f, grad, prep, backend, x, rewrap(unannotated_contexts...)...)
     return nothing
 end
+=#