fix: trim down Mooncake imports (#728)

* fix: trim down Mooncake imports

* Pullback cache

* Ensure prepare gradient works

Author: gdalle

DifferentiationInterface/Project.toml

@@ -62,7 +62,7 @@ ForwardDiff = "0.10.36"
 GTPSA = "1.4.0"
 JuliaFormatter = "1"
 LinearAlgebra = "<0.0.1,1"
-Mooncake = "0.4.83"
+Mooncake = "0.4.88"
 PolyesterForwardDiff = "0.1.2"
 ReverseDiff = "1.15.1"
 SparseArrays = "<0.0.1,1"

DifferentiationInterface/ext/DifferentiationInterfaceMooncakeExt/DifferentiationInterfaceMooncakeExt.jl

@@ -5,14 +5,12 @@ import DifferentiationInterface as DI
 using Mooncake:
     CoDual,
     Config,
-    primal,
-    tangent,
+    prepare_gradient_cache,
+    prepare_pullback_cache,
     tangent_type,
-    value_and_pullback!!,
     value_and_gradient!!,
-    zero_tangent,
-    prepare_pullback_cache,
-    Mooncake
+    value_and_pullback!!,
+    zero_tangent
 
 DI.check_available(::AutoMooncake) = true