Skip to content

feat: implicit differentiation integration #41

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 62 commits into from
Sep 8, 2022
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
62 commits
Select commit Hold shift + click to select a range
6a22b93
feat(third_party/pybind11): switch to 2.9.2
JieRen98 Apr 17, 2022
efc8dd7
Merge branch 'metaopt:main' into dev-implicit
JieRen98 Apr 17, 2022
5cfed82
feat(implicit): add new implicit implementation
JieRen98 May 1, 2022
99a89c5
fix(linear_solver): fix _make_rmatvec
JieRen98 Jul 25, 2022
3155e4a
feat(implicit): allow multi input
JieRen98 Jul 30, 2022
8665895
feat(all): use jax.tree_util.tree_map
JieRen98 Jul 30, 2022
53c31b5
feat(implicit): rename linear_solve.py
JieRen98 Jul 30, 2022
12c30a2
feat(implicit): allow argnums
JieRen98 Aug 9, 2022
f4adb11
fix(implicit): allow has_aux
JieRen98 Aug 9, 2022
76a4a8a
fix(implicit): fix memory leak
JieRen98 Aug 9, 2022
addce03
fix(implicit): add assert
JieRen98 Aug 9, 2022
9de8c4b
fix(implicit): fix cg convergence problem
JieRen98 Aug 12, 2022
1f2bac0
style(implicit): clean codes
JieRen98 Aug 12, 2022
362fbb4
fix(implicit): fix 0-d tensor
JieRen98 Aug 12, 2022
aa29460
fix(implicit): fix 0-d tensor
JieRen98 Aug 12, 2022
309b25d
fix(implicit): fix res is tensor
JieRen98 Aug 13, 2022
ddaa619
fix(implicit): make aux does not require grad
JieRen98 Aug 13, 2022
5af9181
Merge branch 'merge-remote' into dev-implicit
JieRen98 Aug 20, 2022
7d4b1a0
fix(implicit): import implicit functions
JieRen98 Aug 20, 2022
63b5643
style(implicit): reformat
JieRen98 Aug 20, 2022
1114bef
fix(implicit): lint
JieRen98 Aug 20, 2022
abd38fe
Merge branch 'metaopt:main' into dev-implicit
JieRen98 Aug 28, 2022
c055e0d
implicit test
waterhorse1 Aug 29, 2022
074d7b8
docs(implicit): add docs
JieRen98 Aug 29, 2022
9fca772
Merge branch 'metaopt:main' into dev-implicit
JieRen98 Sep 3, 2022
7edfd07
fix(implicit): fix bug
JieRen98 Sep 3, 2022
48f1f88
fix(implicit): fix bug
JieRen98 Sep 3, 2022
f0f3be1
fix(implicit): install jax&jaxopt for test
JieRen98 Sep 4, 2022
060dc5b
fix(implicit): chmod
JieRen98 Sep 4, 2022
33ba3e8
fix(implicit): disable invalid-name
JieRen98 Sep 4, 2022
37ef230
fix(implicit): depend on functorch
JieRen98 Sep 4, 2022
50e5b11
docs(implicit): add description
JieRen98 Sep 4, 2022
24a0e02
fix(implicit): lint bug
JieRen98 Sep 4, 2022
76c6e55
fix(implicit): build bug
JieRen98 Sep 5, 2022
c455ac3
chore: update dependencies
XuehaiPan Sep 5, 2022
2f0ef6c
Merge remote-tracking branch 'upstream/main' into dev-implicit
XuehaiPan Sep 5, 2022
2f9041c
Merge branch 'main' into dev-implicit
XuehaiPan Sep 5, 2022
90f9700
Merge branch 'main' into dev-implicit
XuehaiPan Sep 6, 2022
3185c9d
wip
XuehaiPan Sep 6, 2022
9760fef
wip
XuehaiPan Sep 6, 2022
5fe9129
wip
XuehaiPan Sep 6, 2022
1eace13
wip
XuehaiPan Sep 6, 2022
52a9d60
wip
XuehaiPan Sep 6, 2022
ea24cd1
wip
XuehaiPan Sep 6, 2022
3b00248
wip
XuehaiPan Sep 6, 2022
2e5e882
wip
XuehaiPan Sep 6, 2022
5a26ec2
wip
XuehaiPan Sep 6, 2022
7d79e9e
wip
XuehaiPan Sep 6, 2022
aadfb5d
feat: test other torch specs only once
XuehaiPan Sep 6, 2022
39a1002
fix(implicit): fix scalar bug
JieRen98 Sep 6, 2022
1528f87
fix(implicit): fix scalar bug
JieRen98 Sep 7, 2022
fc56956
fix(implicit): fix solving bug
JieRen98 Sep 7, 2022
75fafcd
fix(implicit): remove unused import
JieRen98 Sep 7, 2022
2dfe754
chore: update type hint
XuehaiPan Sep 7, 2022
3b627f4
fix: update tree inner product
XuehaiPan Sep 7, 2022
028c607
fix: update tree inner product
XuehaiPan Sep 7, 2022
b333d8b
chore: expose pytree
XuehaiPan Sep 7, 2022
02d28b9
style(implicit): change to a compact style
JieRen98 Sep 7, 2022
d8abe9a
new test implicit gradient
waterhorse1 Sep 7, 2022
bde87b1
style(implicit): reformat
JieRen98 Sep 7, 2022
dfd4c30
fix: fix tests
XuehaiPan Sep 8, 2022
ec3b2d3
lint: appease linters
XuehaiPan Sep 8, 2022
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
Prev Previous commit
Next Next commit
fix(implicit): remove unused import
  • Loading branch information
JieRen98 committed Sep 7, 2022
commit 75fafcd14b06dba4ed045e724f6607144e2040cc
1 change: 0 additions & 1 deletion torchopt/_src/linalg.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,6 @@

# pylint: disable=invalid-name

import math
from functools import partial
from typing import Any, Callable, List, Optional, Sequence, Union

Expand Down
1 change: 0 additions & 1 deletion torchopt/_src/linear_solve.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,6 @@
from typing import Callable, Optional

import functorch
import torch

from torchopt._src import linalg
from torchopt._src.typing import TensorTree
Expand Down
pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy