refactor: remove Zygote extension

Author: avik-pal

Project.toml

@@ -45,7 +45,6 @@ NLSolvers = "337daf1e-9722-11e9-073e-8b9effe078ba"
 NLsolve = "2774e3e8-f4cf-5e23-947b-6d7e65073b56"
 SIAMFANLEquations = "084e46ad-d928-497d-ad5e-07fa361a48c4"
 SpeedMapping = "f1835b91-879b-4a3f-a438-e4baacf14412"
-Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [extensions]
 NonlinearSolveBandedMatricesExt = "BandedMatrices"
@@ -57,7 +56,6 @@ NonlinearSolveNLSolversExt = "NLSolvers"
 NonlinearSolveNLsolveExt = "NLsolve"
 NonlinearSolveSIAMFANLEquationsExt = "SIAMFANLEquations"
 NonlinearSolveSpeedMappingExt = "SpeedMapping"
-NonlinearSolveZygoteExt = "Zygote"
 
 [compat]
 ADTypes = "1.9"

ext/NonlinearSolveZygoteExt.jl

@@ -61,9 +61,6 @@ using SparseMatrixColorings: ConstantColoringAlgorithm, GreedyColoringAlgorithm,
 
 const DI = DifferentiationInterface
 
-# Type-Inference Friendly Check for Extension Loading
-is_extension_loaded(::Val) = false
-
 const True = Val(true)
 const False = Val(false)
 

src/NonlinearSolve.jl

@@ -80,7 +80,7 @@ function get_concrete_reverse_ad(
     else
         use_sparse_ad = false
     end
-    ad = if isinplace(prob) || !is_extension_loaded(Val(:Zygote)) # Use Finite Differencing
+    ad = if isinplace(prob) || !DI.check_available(AutoZygote()) # Use Finite Differencing
         use_sparse_ad ? AutoSparse(AutoFiniteDiff()) : AutoFiniteDiff()
     else
         use_sparse_ad ? AutoSparse(AutoZygote()) : AutoZygote()

src/internal/helpers.jl

@@ -52,12 +52,6 @@
     sol = solve(prob_brusselator_2d_sparse, NewtonRaphson(); abstol = 1e-8)
     @test norm(sol.resid, Inf) < 1e-8
 
-    prob_brusselator_2d_sparse = NonlinearProblem(
-        NonlinearFunction(brusselator_2d_loop; sparsity = SymbolicsSparsityDetector()),
-        u0, p)
-    sol = solve(prob_brusselator_2d_sparse, NewtonRaphson(); abstol = 1e-8)
-    @test norm(sol.resid, Inf) < 1e-8
-
     # Deprecated
     sol = solve(prob_brusselator_2d,
         NewtonRaphson(autodiff = AutoSparse(AutoFiniteDiff())); abstol = 1e-8)

test/misc/bruss_tests.jl

@@ -17,8 +17,7 @@ end
 @testitem "Explicit Imports" tags=[:misc] begin
     using NonlinearSolve, ADTypes, SimpleNonlinearSolve, SciMLBase
     import BandedMatrices, FastLevenbergMarquardt, FixedPointAcceleration,
-           LeastSquaresOptim, MINPACK, NLsolve, NLSolvers, SIAMFANLEquations, SpeedMapping,
-           Zygote
+           LeastSquaresOptim, MINPACK, NLsolve, NLSolvers, SIAMFANLEquations, SpeedMapping
 
     using ExplicitImports