Skip to content

[graph_trainer] Add torch.no_grad() and graph-based SAC to traced execution #2766

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

Merged
tugsbayasgalan merged 90 commits into from
Apr 16, 2026
Merged

[graph_trainer] Add torch.no_grad() and graph-based SAC to traced execution #2766

Show file tree
Hide file tree
Changes from 14 commits
Commits
Show all changes
90 commits
Select commit Hold shift + click to select a range
3d04ab9
[graph_trainer] Replace trace_module/run_traced_module with aot_funct…
tugsbayasgalan Mar 30, 2026
da526a8
Update on "[graph_trainer] Replace trace_module/run_traced_module wit…
tugsbayasgalan Mar 30, 2026
3394aaf
Update on "[graph_trainer] Replace trace_module/run_traced_module wit…
tugsbayasgalan Mar 31, 2026
8f276ba
Update on "[graph_trainer] Replace trace_module/run_traced_module wit…
tugsbayasgalan Mar 31, 2026
4f56847
Update on "[graph_trainer] Replace trace_module/run_traced_module wit…
tugsbayasgalan Mar 31, 2026
155d120
Update on "[graph_trainer] Replace trace_module/run_traced_module wit…
tugsbayasgalan Mar 31, 2026
49bd670
[graph_trainer] Add remat pass and torch.no_grad() execution to minim…
tugsbayasgalan Mar 31, 2026
d7e80e0
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Mar 31, 2026
2ebcc94
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Mar 31, 2026
92f7588
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Mar 31, 2026
6960a36
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Mar 31, 2026
bfb303b
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Mar 31, 2026
0b66ad1
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Mar 31, 2026
4dfbaf8
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Mar 31, 2026
cc97983
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Mar 31, 2026
10de659
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Mar 31, 2026
0783db3
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Mar 31, 2026
4b4ec15
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Mar 31, 2026
4b48a21
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 1, 2026
03b648f
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 1, 2026
84c908a
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 1, 2026
4e9d810
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 1, 2026
d538249
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 1, 2026
b3daa48
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 1, 2026
201ed86
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 1, 2026
055e0ce
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 1, 2026
fd4bd77
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 2, 2026
f0eb787
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 2, 2026
803cc90
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 2, 2026
24a1be0
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 2, 2026
6559ba6
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 3, 2026
4c1d0d2
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 3, 2026
3036bae
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 4, 2026
467d42e
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 4, 2026
386810a
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 4, 2026
e3867cb
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 4, 2026
933726a
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 6, 2026
ef5babd
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 6, 2026
6580931
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 6, 2026
bc213b6
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 6, 2026
8bd77b7
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 6, 2026
0ca9c95
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 6, 2026
9693ffe
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 6, 2026
f17af31
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 6, 2026
eba5038
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 6, 2026
ae36cae
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 6, 2026
f1ae755
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 6, 2026
04f2ac4
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 6, 2026
3bcc838
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 6, 2026
83dcd2b
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 6, 2026
9b92650
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 7, 2026
dfca436
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 7, 2026
01d002a
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 8, 2026
3726f86
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 8, 2026
85526b2
Update base for Update on "[graph_trainer] Add remat pass and torch.n…
tugsbayasgalan Apr 8, 2026
a9b01f9
Update on "[graph_trainer] Add remat pass and torch.no_grad() executi…
tugsbayasgalan Apr 8, 2026
fa35998
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 8, 2026
0e44929
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 8, 2026
0ea91a0
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 9, 2026
19afae3
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 9, 2026
9e42167
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 10, 2026
a2f3343
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 10, 2026
13cf04c
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 12, 2026
246b992
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 12, 2026
32942cd
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 12, 2026
775dfe7
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 12, 2026
a838dc2
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 13, 2026
1618ddd
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 13, 2026
7bb5d54
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 13, 2026
fdaa93c
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 13, 2026
3a64ffd
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 13, 2026
339194e
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 13, 2026
404b1a3
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 14, 2026
202f0cb
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 14, 2026
5a02e95
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 14, 2026
a8b13f8
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 14, 2026
b28baa4
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 14, 2026
4c620f5
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 14, 2026
e7e251c
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 14, 2026
1267aae
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 14, 2026
64bf2c7
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 15, 2026
87b7abc
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 15, 2026
d28765b
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 15, 2026
1147da4
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 15, 2026
7f57f92
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 15, 2026
24d5218
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 15, 2026
723e0e4
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 15, 2026
d989706
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 15, 2026
4bdc488
Update base for Update on "[graph_trainer] Add torch.no_grad() and gr…
tugsbayasgalan Apr 16, 2026
c60367c
Update on "[graph_trainer] Add torch.no_grad() and graph-based SAC to…
tugsbayasgalan Apr 16, 2026
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
Loading
Loading