diff --git a/docs/Project.toml b/docs/Project.toml index 568495207..ddbb79596 100644 --- a/docs/Project.toml +++ b/docs/Project.toml @@ -39,7 +39,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" [compat] CSV = "0.10" CUDA = "5" -ComponentArrays = "0.13, 0.14, 0.15" +ComponentArrays = "0.15" DataDeps = "0.7" DataFrames = "1" DiffEqFlux = "3" @@ -49,27 +49,27 @@ Documenter = "1" Flux = "0.14" ForwardDiff = "0.10" IterTools = "1" -LinearAlgebra = "<0.0.1, 1" +LinearAlgebra = "1" Lux = "0.5.5" LuxCUDA = "0.3" MLDataUtils = "0.5" MLDatasets = "0.7" MLUtils = "0.4" NNlib = "0.9" -Optimisers = "0.2, 0.3" +Optimisers = "0.3" Optimization = "3.9" OptimizationOptimJL = "0.2" OptimizationOptimisers = "0.2" OptimizationPolyalgorithms = "0.2" OrdinaryDiffEq = "6.31" Plots = "1.36" -Printf = "<0.0.1, 1" -Random = "<0.0.1, 1" +Printf = "1" +Random = "1" ReverseDiff = "1.14" -SciMLBase = "1.72, 2" +SciMLBase = "2" SciMLSensitivity = "7.11" StableRNGs = "1" Statistics = "1" StochasticDiffEq = "6.56" -Test = "<0.0.1, 1" +Test = "1" Zygote = "0.6.62" diff --git a/src/DiffEqFlux.jl b/src/DiffEqFlux.jl index 06e9c59b0..cd6c451f5 100644 --- a/src/DiffEqFlux.jl +++ b/src/DiffEqFlux.jl @@ -19,16 +19,6 @@ import Lux.Experimental: StatefulLuxLayer @reexport using ADTypes, Lux -# FIXME: Type Piracy -function CRC.rrule(::Type{Tridiagonal}, dl, d, du) - y = Tridiagonal(dl, d, du) - @views function ∇Tridiagonal(∂y) - return (NoTangent(), diag(∂y[2:end, 1:(end - 1)]), diag(∂y), - diag(∂y[1:(end - 1), 2:end])) - end - return y, ∇Tridiagonal -end - include("ffjord.jl") include("neural_de.jl") include("spline_layer.jl")