Skip to content
This repository was archived by the owner on Aug 25, 2025. It is now read-only.
Merged

V2 #90

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
38 commits
Select commit Hold shift + click to select a range
57fbfae
Fresh attempt at DI integration
Vaibhavdixit02 May 31, 2024
e771c6c
DI to latest
Vaibhavdixit02 May 31, 2024
3b41754
Flesh out latest ideas
Vaibhavdixit02 Jun 6, 2024
3951c88
EnzymeExt and fix mode checking
Vaibhavdixit02 Jun 7, 2024
ad7ca08
Use the changes from main in enzyme
Vaibhavdixit02 Jul 9, 2024
4fe1f54
Flesh out sparsity and secondorder things more, change the extension …
Vaibhavdixit02 Jul 15, 2024
f0b8401
bump adtypes
Vaibhavdixit02 Jul 15, 2024
48ad2f4
Use only the dense_ad in gradient and misc ups
Vaibhavdixit02 Jul 15, 2024
b9479d0
Use global sparsity detector
Vaibhavdixit02 Jul 16, 2024
517c061
format
Vaibhavdixit02 Jul 16, 2024
7beebae
handle reinitcache dispatches better
Vaibhavdixit02 Jul 17, 2024
27cd0d4
try to get downstream running
Vaibhavdixit02 Jul 19, 2024
35bd067
hvpextras preparation with random v
Vaibhavdixit02 Jul 19, 2024
c1a5e1f
fix sparse adtype passed to hvp
Vaibhavdixit02 Jul 19, 2024
1325478
Merge pull request #54 from SciML/Diattempt2
Vaibhavdixit02 Jul 20, 2024
bfb28b6
Move to no closures and no args
Vaibhavdixit02 Jul 23, 2024
591e87c
cons_oop oops
Vaibhavdixit02 Jul 23, 2024
862be58
clarify import
Vaibhavdixit02 Jul 23, 2024
eb74297
generate a sparse lagrange hessian and some more things
Vaibhavdixit02 Jul 31, 2024
6d4f6d7
hessian of lagrangian
Vaibhavdixit02 Aug 2, 2024
9813fd4
vjp and jvp and jacobian based on dimensions
Vaibhavdixit02 Aug 4, 2024
f17c54c
jvp vjp with DI
Vaibhavdixit02 Aug 4, 2024
e79d4cb
Inplace all passing
Vaibhavdixit02 Aug 5, 2024
fe1b709
Oop improvements
Vaibhavdixit02 Aug 6, 2024
65781ba
Merge pull request #80 from SciML/rethinkAPI
Vaibhavdixit02 Aug 6, 2024
142f8ac
boolean switches for oracles
Vaibhavdixit02 Aug 7, 2024
8fb865d
bool switch everywhere
Vaibhavdixit02 Aug 16, 2024
2bad384
more of the switches, and in tests too
Vaibhavdixit02 Aug 16, 2024
2792e2b
update SpraseMatrixColoring v0.4
Vaibhavdixit02 Aug 16, 2024
bbfdf96
Merge pull request #85 from SciML/rethinkAPI
Vaibhavdixit02 Aug 18, 2024
c161954
Hack p to serve as data arg and implement stochastic gradient oracle
Vaibhavdixit02 Aug 22, 2024
3e04244
use master project.toml
Vaibhavdixit02 Aug 22, 2024
36963a4
format
Vaibhavdixit02 Aug 22, 2024
602227c
Merge pull request #89 from SciML/stochgrad
Vaibhavdixit02 Aug 22, 2024
358b1e9
Merge branch 'main' into v2
Vaibhavdixit02 Aug 22, 2024
ec157c3
Update NoAD dispatch
Vaibhavdixit02 Aug 22, 2024
ec3067b
Traits based boolean switching in instantiate_function call
Vaibhavdixit02 Aug 22, 2024
54162cf
format
Vaibhavdixit02 Aug 23, 2024
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/Downstream.yml
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ jobs:
julia-version: [1]
os: [ubuntu-latest]
package:
- {user: SciML, repo: Optimization.jl, group: Optimization}
- {user: SciML, repo: Optimization.jl, group: All}

steps:
- uses: actions/checkout@v4
Expand Down
15 changes: 7 additions & 8 deletions Project.toml
Original file line number Diff line number Diff line change
Expand Up @@ -6,13 +6,17 @@ version = "1.5.0"
[deps]
ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
DifferentiationInterface = "a0c0ee7d-e4b9-4e03-894e-1c5f64a51d63"
DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
FastClosures = "9aa1b823-49e4-5ca5-8b0f-3971ec8bab6a"
LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
PDMats = "90014a1f-27ba-587c-ab20-58faa44d9150"
Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
Requires = "ae029012-a4dd-5104-9daa-d747884805df"
SciMLBase = "0bca4576-84f4-4d90-8ffe-ffa030f20462"
SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
SparseConnectivityTracer = "9f842d2f-2579-4b1d-911e-f412cf18a3f5"
SparseMatrixColorings = "0a514795-09f3-496d-8182-132a7b665d35"
SymbolicAnalysis = "4297ee4d-0239-47d8-ba5d-195ecdf594fe"
SymbolicIndexingInterface = "2efcf032-c050-4f8e-a9bb-153293bab1f5"
Symbolics = "0c5d862f-8b57-4792-8d23-62f2024744c7"
Expand All @@ -23,8 +27,6 @@ FiniteDiff = "6a86dc24-6348-571c-b903-95158fe2bd41"
ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
ModelingToolkit = "961ee093-0014-501f-94e3-6117800e7a78"
ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
SparseDiffTools = "47a9eef4-7e08-11e9-0b38-333d64bd3804"
Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"

[extensions]
Expand All @@ -33,13 +35,12 @@ OptimizationFiniteDiffExt = "FiniteDiff"
OptimizationForwardDiffExt = "ForwardDiff"
OptimizationMTKExt = "ModelingToolkit"
OptimizationReverseDiffExt = "ReverseDiff"
OptimizationSparseDiffExt = ["SparseDiffTools", "ReverseDiff"]
OptimizationTrackerExt = "Tracker"
OptimizationZygoteExt = "Zygote"

[compat]
ADTypes = "1.3"
ADTypes = "1.5"
ArrayInterface = "7.6"
DifferentiationInterface = "0.5"
DocStringExtensions = "0.9"
Enzyme = "0.12.12"
FiniteDiff = "2.12"
Expand All @@ -51,16 +52,14 @@ Reexport = "1.2"
Requires = "1"
ReverseDiff = "1.14"
SciMLBase = "2"
SparseDiffTools = "2.14"
SymbolicAnalysis = "0.3"
SymbolicIndexingInterface = "0.3"
Symbolics = "5.12, 6"
Tracker = "0.2.29"
Zygote = "0.6.67"
julia = "1.10"

[extras]
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

[targets]
test = ["Test"]
test = ["Test"]
Loading