Merge pull request #39 from SciML/ap/up_lux

chore: force install latest versions

Project.toml

@@ -29,7 +29,7 @@ SafeTestsets = "0.1"
 SciMLStructures = "1.1.0"
 StableRNGs = "1"
 SymbolicIndexingInterface = "0.3.15"
-Symbolics = "5.27, 6.0"
+Symbolics = "6.9"
 Test = "1.10"
 julia = "1.10"
 

docs/Project.toml

@@ -14,7 +14,7 @@ ModelingToolkitNeuralNets = "f162e290-f571-43a6-83d9-22ecc16da15f"
 
 [compat]
 Documenter = "1.3"
-Lux = "0.5.32"
+Lux = "1"
 ModelingToolkit = "9.9"
 ModelingToolkitStandardLibrary = "2.7"
 Optimization = "3.24"

src/utils.jl

@@ -1,6 +1,6 @@
 function multi_layer_feed_forward(input_length, output_length; width::Int = 5,
-        depth::Int = 1, activation = tanh, disable_optimizations = false)
+        depth::Int = 1, activation = tanh)
     Lux.Chain(Lux.Dense(input_length, width, activation),
         [Lux.Dense(width, width, activation) for _ in 1:(depth)]...,
-        Lux.Dense(width, output_length); disable_optimizations)
+        Lux.Dense(width, output_length))
 end