LuxDL · avik-pal · Jan 30, 2026 · Jan 30, 2026 · Jan 30, 2026 · Jan 30, 2026
diff --git a/.buildkite/testing.yml b/.buildkite/testing.yml
@@ -1,12 +1,12 @@
 steps:
   - group: ":julia: CUDA GPU"
     steps:
-      - label: ":julia: Julia {{matrix.julia}} + CUDA GPU + {{matrix.group}} + Reactant: {{matrix.reactant}}"
+      - label: ":julia: Julia {{matrix.julia}} + CUDA GPU + {{matrix.group}}"
         plugins:
           - JuliaCI/julia#v1:
               version: "{{matrix.julia}}"
           - JuliaCI/julia-test#v1:
-              test_args: "BACKEND_GROUP=CUDA BOLTZ_TEST_GROUP={{matrix.group}} BOLTZ_TEST_REACTANT={{matrix.reactant}}"
+              test_args: "{{matrix.group}}"
           - JuliaCI/julia-coverage#v1:
               codecov: true
               dirs:
@@ -22,19 +22,19 @@ steps:
             julia:
               - "1"
             group:
-              - "all"
-            reactant:
-              - "true"
-              - "false"
+              - "layers"
+              - "vision"
+              - "piml"
+              - "core"
 
   # - group: ":julia: AMD GPU"
   #   steps:
-  #     - label: ":julia: Julia: {{matrix.julia}} + AMD GPU + {{matrix.group}} + Reactant: {{matrix.reactant}}"
+  #     - label: ":julia: Julia: {{matrix.julia}} + AMD GPU + {{matrix.group}}"
   #       plugins:
   #         - JuliaCI/julia#v1:
   #             version: "{{matrix.julia}}"
   #         - JuliaCI/julia-test#v1:
-  #             test_args: "BACKEND_GROUP=AMDGPU BOLTZ_TEST_GROUP={{matrix.group}} BOLTZ_TEST_REACTANT={{matrix.reactant}}"
+  #             test_args: "{{matrix.group}}"
   #         - JuliaCI/julia-coverage#v1:
   #             codecov: true
   #             dirs:
@@ -55,9 +55,10 @@ steps:
   #           julia:
   #             - "1"
   #           group:
-  #             - "all"
-  #           reactant:
-  #             - "false"
+  #             - "layers"
+  #             - "vision"
+  #             - "piml"
+  #             - "core"
 
 env:
   SECRET_CODECOV_TOKEN: "gZlC/IAmeUJehhP5mP2QuUV5a1qV61cvo4PUCLkA9vVkt3x6wgD6fTZmCm+f+gHkmkssFxX+q2h1Ud00XXc75H2LrjyR/cDTIthcO46BBOidYocv/U0gfhp6uT2IZ9fi+ryFfTVVpZ0RIUGmDTj0O/b5qt4oaTriAArLAq6mMipbIR9YCz7ZD/hWQXx8oDeAbnDpwQaddwPyhJhz95nayknOpuJj+ClaVOxgsLGZc3ZWiTj1QxkXBNwxLD2ALeG16Qxs9h7eK87sdcbWeTihvJ6OooARgpoVJAa2pJCFYOGy4Bh07c0VTZmicN2M3GIi74Y5T1PWNaz7nGeANO5Pow==;U2FsdGVkX1843DHkbGWCV9PArLBw0rNqmdy56VOTRNTifBSpkC796Oez1lMFU+yDtkElbcrRSIlS5hRFqpsaFA=="
diff --git a/.github/dependabot.yml b/.github/dependabot.yml
@@ -1,7 +1,17 @@
 # https://docs.github.com/github/administering-a-repository/configuration-options-for-dependency-updates
 version: 2
+enable-beta-ecosystems: true
 updates:
   - package-ecosystem: "github-actions"
     directory: "/" # Location of package manifests
     schedule:
       interval: "weekly"
+
+  - package-ecosystem: "julia"
+    directories:
+      # Location of Julia projects
+      - "/"
+      - "/examples/GettingStarted"
+      - "/examples/SymbolicOptimalControl"
+    schedule:
+      interval: "weekly"
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -29,13 +29,16 @@ jobs:
         os:
           - ubuntu-latest
         group:
-          - "all"
+          - "layers"
+          - "vision"
+          - "piml"
+          - "core"
     uses: LuxDL/Lux.jl/.github/workflows/CommonCI.yml@main
     with:
       julia_version: ${{ matrix.version }}
       os: ${{ matrix.os }}
       project: "."
-      test_args: "BACKEND_GROUP=CPU BOLTZ_TEST_GROUP=${{ matrix.group }}"
+      test_args: "${{ matrix.group }}"
 
   downgrade:
     uses: LuxDL/Lux.jl/.github/workflows/CommonCI.yml@main
@@ -44,4 +47,4 @@ jobs:
       os: ubuntu-latest
       project: "."
       downgrade_testing: true
-      test_args: "BACKEND_GROUP=CPU BOLTZ_TEST_GROUP=all"
+      test_args: ""
diff --git a/.github/workflows/CompatHelper.yml b/.github/workflows/CompatHelper.yml
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "Boltz"
 uuid = "4544d5e4-abc5-4dea-817f-29e4c205d9c8"
 authors = ["Avik Pal <avikpal@mit.edu> and contributors"]
-version = "1.7.1"
+version = "1.8.0"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
@@ -56,7 +56,7 @@ BoltzTrackerExt = "Tracker"
 BoltzZygoteExt = "Zygote"
 
 [compat]
-ADTypes = "1.10"
+ADTypes = "1.15"
 ArgCheck = "2.3"
 Artifacts = "1.10, 1"
 ChainRulesCore = "1.25.1"
@@ -73,7 +73,7 @@ LazyArtifacts = "1.10"
 Lux = "1.21.2"
 LuxCore = "1.2"
 LuxLib = "1.11"
-MLDataDevices = "1.11.2"
+MLDataDevices = "1.17.3"
 Markdown = "1.10"
 Metalhead = "0.9.5"
 NNlib = "0.9.30"
@@ -84,11 +84,14 @@ ReactantCore = "0.1.15"
 Reexport = "1.2.2"
 ReverseDiff = "1.16.1"
 SafeTensors = "1.2"
-Scratch = "1.2"
+Scratch = "1.3"
 Setfield = "1.1.2"
 Static = "1.1.1"
 Statistics = "1.10"
 Tracker = "0.2.38"
 WeightInitializers = "1"
 Zygote = "0.7.7"
 julia = "1.10"
+
+[workspace]
+projects = ["test", "docs"]
diff --git a/examples/GettingStarted/Project.toml b/examples/GettingStarted/Project.toml
@@ -5,6 +5,9 @@ Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
 Metalhead = "dbeba491-748d-5e0e-a39e-b530a07fa0cc"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 
+[sources]
+Boltz = {path = "../.."}
+
 [compat]
 Boltz = "1"
 JLD2 = "0.5, 0.6"

diff --git a/examples/SymbolicOptimalControl/Project.toml b/examples/SymbolicOptimalControl/Project.toml
@@ -17,19 +17,22 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 SymbolicRegression = "8254be44-1295-4e6a-a16d-46603ac705cb"
 SymbolicUtils = "d1185830-fcd6-423d-90d6-eec64667417b"
 
+[sources]
+Boltz = {path = "../.."}
+
 [compat]
 Boltz = "1"
 CairoMakie = "0.12, 0.13, 0.14, 0.15"
 ComponentArrays = "0.15.11"
 DynamicExpressions = "1.10, 2"
 Latexify = "0.16.2"
 Lux = "1.21.2"
-MLJ = "0.20.3, 0.21"
-Optimization = "4"
+MLJ = "0.20.3, 0.21, 0.22"
+Optimization = "4, 5"
 OptimizationOptimJL = "0.4"
 OptimizationOptimisers = "0.3.2"
 OrdinaryDiffEqVerner = "1"
 SciMLSensitivity = "7.57"
 Statistics = "1.10"
 SymbolicRegression = "1"
-SymbolicUtils = "3"
+SymbolicUtils = "3, 4"
diff --git a/examples/SymbolicOptimalControl/main.jl b/examples/SymbolicOptimalControl/main.jl
@@ -32,17 +32,9 @@
 
 # ## Package Imports
 
-using Lux,
-    Boltz,
-    ComponentArrays,
-    OrdinaryDiffEqVerner,
-    Optimization,
-    OptimizationOptimJL,
-    OptimizationOptimisers,
-    SciMLSensitivity,
-    Statistics,
-    Printf,
-    Random
+using Lux, Boltz, ComponentArrays, Statistics, Printf, Random
+using Optimization,
+    OptimizationOptimJL, OptimizationOptimisers, SciMLSensitivity, OrdinaryDiffEqVerner
 using DynamicExpressions, SymbolicRegression, MLJ, SymbolicUtils, Latexify
 using CairoMakie
 

diff --git a/src/initialize.jl b/src/initialize.jl
@@ -1,6 +1,5 @@
 module InitializeModels
 
-using ArgCheck: @argcheck
 using Functors: fmap
 using Random: Random, AbstractRNG
 

diff --git a/src/piml/PIML.jl b/src/piml/PIML.jl
@@ -1,6 +1,5 @@
 module PIML
 
-using ArgCheck: @argcheck
 using Compat: @compat
 using ConcreteStructs: @concrete
 using Random: Random, AbstractRNG

diff --git a/test/Project.toml b/test/Project.toml
@@ -1,6 +1,6 @@
 [deps]
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
-CPUSummary = "2a0fbf3d-bb9c-48f3-b0a9-814d99fd7ab9"
+Boltz = "4544d5e4-abc5-4dea-817f-29e4c205d9c8"
 ComponentArrays = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
 DataInterpolations = "82cc6244-b520-54b8-b5a6-8a565e85f1d0"
 Downloads = "f43a241f-c20a-4ad4-852c-f6b1247861c6"
@@ -11,25 +11,25 @@ ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
 JLD2 = "033835bb-8acc-5ee8-8aae-3f567f8a3819"
 Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
-LuxCUDA = "d0bbae9a-e099-4d5b-a835-1c6931763bda"
 LuxLib = "82251201-b29d-42c6-8e01-566dec8acb11"
 LuxTestUtils = "ac9de150-d08f-4546-94fb-7472b5760531"
 MLDataDevices = "7e8f7934-dd98-4c1a-8fe8-92b47a384d40"
 Metalhead = "dbeba491-748d-5e0e-a39e-b530a07fa0cc"
 NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
+ParallelTestRunner = "d3525ed8-44d0-4b2c-a655-542cee43accc"
 Pickle = "fbb45041-c46e-462f-888f-7c521cafbc2c"
 Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
-ReTestItems = "817f1d60-ba6b-4fd5-9520-3cf149f6a823"
 Reactant = "3c362404-f566-11ee-1572-e11a4b42c853"
-Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
+Setfield = "efcf1570-3423-57d1-acb7-fd33fddbac46"
 StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
-Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
+
+[sources]
+Boltz = {path = ".."}
 
 [compat]
 Aqua = "0.8.7"
-CPUSummary = "0.2"
 ComponentArrays = "0.15.16"
 DataInterpolations = "8"
 Downloads = "1.6"
@@ -40,18 +40,15 @@ ForwardDiff = "1"
 GPUArraysCore = "0.1.6, 0.2"
 JLD2 = "0.5, 0.6"
 Lux = "1.21.2"
-LuxCUDA = "0.3.3"
 LuxLib = "1"
 LuxTestUtils = "2"
 MLDataDevices = "1.11.2"
 Metalhead = "0.9.5"
 NNlib = "0.9.30"
+ParallelTestRunner = "2.1"
 Pickle = "0.3.5"
 Pkg = "1.10"
 Random = "1.10"
-ReTestItems = "1.24.0"
 Reactant = "0.2.97"
-Reexport = "1.2.2"
 StableRNGs = "1.0.2"
 Test = "1.10"
-Zygote = "0.7.7"
diff --git a/test/qa_tests.jl → test/core/qa.jl b/test/qa_tests.jl → test/core/qa.jl
@@ -1,15 +1,11 @@
-@testitem "Aqua: Quality Assurance" tags = [:others] begin
-    using Aqua
+using Lux, Aqua, ExplicitImports, Boltz, Test
 
+@testset "Aqua: Quality Assurance" begin
     Aqua.test_all(Boltz; ambiguities=false)
     Aqua.test_ambiguities(Boltz; recursive=false)
 end
 
-@testitem "Explicit Imports: Quality Assurance" tags = [:others] begin
-    using Lux: Lux
-    using Zygote: Zygote # Load all trigger packages
-    using ExplicitImports
-
+@testset "Explicit Imports: Quality Assurance" begin
     @test check_no_implicit_imports(Boltz; skip=(Base, Core, Lux)) === nothing
     @test check_no_stale_explicit_imports(Boltz) === nothing
     @test check_no_self_qualified_accesses(Boltz) === nothing