Skip to content
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.

Sign up for GitHub

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

Dev add jvp jacobian hessian #10412

Merged
merged 35 commits into from
Jan 23, 2024
Merged

Dev add jvp jacobian hessian #10412

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
35 commits
Select commit Hold shift + click to select a range
fa04b47
add jvp interface and test
lihuizhao Jan 2, 2024
57ec099
add jvp interface and test
lihuizhao Jan 2, 2024
256d75a
add annotation information
lihuizhao Jan 2, 2024
cc7b43d
add jacobian
lihuizhao Jan 4, 2024
4f4c1d5
add vhp
lihuizhao Jan 4, 2024
49dfb97
add vhp test
lihuizhao Jan 4, 2024
f26d539
add hvp
lihuizhao Jan 4, 2024
65d90ce
add hvp test and fix vhp test
lihuizhao Jan 4, 2024
c6b8298
fix jacobian interface when the condition is strategy=reverse-mode an…
lihuizhao Jan 4, 2024
b5bac0e
fix jacobian interface when the condition is strategy=reverse-mode an…
lihuizhao Jan 5, 2024
5906339
add jacobian tests and hessian tests
lihuizhao Jan 10, 2024
e69a37c
format code
lihuizhao Jan 10, 2024
4bbb368
code format
lihuizhao Jan 10, 2024
030dc61
change jvp doc
lihuizhao Jan 10, 2024
6b8db1f
add jvp/jacobian/hessian/vhp/hvp to autograd.rst
lihuizhao Jan 10, 2024
233d423
add annotation
lihuizhao Jan 10, 2024
f54c58b
remove _batched_autograd_grad() function
lihuizhao Jan 12, 2024
81fe941
format
lihuizhao Jan 15, 2024
6efac49
Merge branch 'master' into dev_add_jvp_jacobian_hessian
lihuizhao Jan 15, 2024
6ab73d6
fix
lihuizhao Jan 16, 2024
ab874fb
Remove extra code
lihuizhao Jan 16, 2024
615751e
change torch
lihuizhao Jan 20, 2024
cc2a98d
change torch.randn to random_tensor
lihuizhao Jan 22, 2024
4c0ab1c
add autotest
lihuizhao Jan 22, 2024
2d547ff
format
lihuizhao Jan 22, 2024
db81d60
Merge branch 'master' into dev_add_jvp_jacobian_hessian
marigoold Jan 22, 2024
176b9fb
Merge branch 'master' into dev_add_jvp_jacobian_hessian
marigoold Jan 23, 2024
7b09742
Remove the hessian test
lihuizhao Jan 23, 2024
2616c22
Merge branch 'master' of https://github.com/Oneflow-Inc/oneflow into …
lihuizhao Jan 23, 2024
1bcd8f9
Merge branch 'dev_add_jvp_jacobian_hessian' of https://github.com/One…
lihuizhao Jan 23, 2024
16399d8
auto format by CI
oneflow-ci-bot Jan 23, 2024
848258d
format
lihuizhao Jan 23, 2024
3531c2c
Merge branch 'dev_add_jvp_jacobian_hessian' of https://github.com/One…
lihuizhao Jan 23, 2024
edce8d1
remove test_jacobian
lihuizhao Jan 23, 2024
0918c85
format
lihuizhao Jan 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
7 changes: 6 additions & 1 deletion docs/source/autograd.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -93,4 +93,9 @@ functional
:toctree: generated
:nosignatures:

vjp
vjp
jvp
jacobian
hessian
vhp
hvp
7 changes: 7 additions & 0 deletions python/oneflow/autograd/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
is_grad_enabled,
no_grad,
)
from oneflow.autograd.functional import vjp, jvp, jacobian, hessian, hvp, vhp
from . import graph

__all__ = [
Expand All @@ -34,4 +35,10 @@
"inference_mode",
"is_grad_enabled",
"no_grad",
"vjp",
"jvp",
"jacobian",
"hessian",
"hvp",
"vhp",
]
Loading