add old versions compat

Author: prbzrg

Project.toml

@@ -11,7 +11,6 @@ DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
 DifferentiationInterface = "a0c0ee7d-e4b9-4e03-894e-1c5f64a51d63"
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 DistributionsAD = "ced4e74d-a319-5a8a-b0ac-84af2272839c"
-FastBroadcast = "7034ab61-46d4-4ed7-9d0f-46aef9175898"
 FillArrays = "1a297f60-69ca-5386-bcde-b61e274b549b"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
@@ -24,7 +23,6 @@ NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
 Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
 OptimizationOptimisers = "42dfb2eb-d2b4-4451-abcd-913932933ac1"
 OrdinaryDiffEqAdamsBashforthMoulton = "89bda076-bce5-4f1c-845f-551c83cdda9a"
-Polyester = "f517fe37-dbe3-4b94-8317-1923a5111588"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 SciMLBase = "0bca4576-84f4-4d90-8ffe-ffa030f20462"
 SciMLLogging = "a6db7da4-7206-11f0-1eab-35f2a5dbe1d1"
@@ -42,7 +40,6 @@ DataFrames = "1"
 DifferentiationInterface = "0.7"
 Distributions = "0.25"
 DistributionsAD = "0.6"
-FastBroadcast = "1"
 FillArrays = "1"
 LinearAlgebra = "1"
 Lux = "1"
@@ -54,10 +51,9 @@ MLUtils = "0.4"
 NNlib = "0.9"
 Optimisers = "0.4"
 OptimizationOptimisers = "0.3"
-OrdinaryDiffEqAdamsBashforthMoulton = "2"
-Polyester = "0.7"
+OrdinaryDiffEqAdamsBashforthMoulton = "1, 2"
 Random = "1"
-SciMLBase = "3"
+SciMLBase = "2, 3"
 SciMLLogging = "1, 2"
 SciMLSensitivity = "7"
 ScientificTypesBase = "3"

examples/Project.toml

@@ -5,14 +5,12 @@ ContinuousNormalizingFlows = "00b1973d-5b2e-40bf-8604-5c9c1d8f50ac"
 DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
 Distances = "b4f34e82-e78d-54a5-968a-f98e89d6e8f7"
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
-FastBroadcast = "7034ab61-46d4-4ed7-9d0f-46aef9175898"
 Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
 Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
 MLDataDevices = "7e8f7934-dd98-4c1a-8fe8-92b47a384d40"
 MLJBase = "a7f614a8-145f-11e9-1d2a-a57a1082229d"
 OptimizationOptimisers = "42dfb2eb-d2b4-4451-abcd-913932933ac1"
 OrdinaryDiffEqAdamsBashforthMoulton = "89bda076-bce5-4f1c-845f-551c83cdda9a"
-Polyester = "f517fe37-dbe3-4b94-8317-1923a5111588"
 SciMLLogging = "a6db7da4-7206-11f0-1eab-35f2a5dbe1d1"
 SciMLSensitivity = "1ed8b502-d754-442c-8d5d-10ac956f44a1"
 TerminalLoggers = "5d786b92-1e48-4d6f-9151-6b4477ca9bed"

examples/usage.jl

@@ -20,8 +20,6 @@ n_hidden = n_in * 4
 using ContinuousNormalizingFlows,
     Lux,
     OrdinaryDiffEqAdamsBashforthMoulton,
-    FastBroadcast,
-    Polyester,
     SciMLLogging,
     SciMLSensitivity,
     ADTypes,
@@ -55,7 +53,7 @@ icnf = ICNF(;
         maxiters = typemax(Int),
         reltol = 1.0e-4,
         abstol = 1.0e-8,
-        alg = VCABM(; thread = Threaded()),
+        alg = VCABM(),
         sensealg = QuadratureAdjoint(;
             autodiff = true,
             autojacvec = ZygoteVJP(),

src/ContinuousNormalizingFlows.jl

@@ -7,7 +7,6 @@ import ADTypes,
     DifferentiationInterface,
     Distributions,
     DistributionsAD,
-    FastBroadcast,
     FillArrays,
     LinearAlgebra,
     Lux,
@@ -20,7 +19,6 @@ import ADTypes,
     Optimisers,
     OptimizationOptimisers,
     OrdinaryDiffEqAdamsBashforthMoulton,
-    Polyester,
     Random,
     SciMLBase,
     SciMLLogging,

src/core/icnf.jl

@@ -86,9 +86,7 @@ function ICNF(;
         maxiters = typemax(Int),
         reltol = convert(data_type, 1.0e-4),
         abstol = convert(data_type, 1.0e-8),
-        alg = OrdinaryDiffEqAdamsBashforthMoulton.VCABM(;
-            thread = FastBroadcast.Threaded(),
-        ),
+        alg = OrdinaryDiffEqAdamsBashforthMoulton.VCABM(),
         sensealg = SciMLSensitivity.QuadratureAdjoint(;
             autodiff = true,
             autojacvec = ifelse(