fix(autobatching): detach states to stop UMA autograd-graph leak (swap loop + chunked init)#590

Merged

CompRhys merged 7 commits into

TorchSim:mainfrom

niklashoelter:fix/uma-inflight-energy-graph-leak

Jul 21, 2026

Commits on Jul 4, 2026

fix(autobatching): detach completed states to stop UMA forward-graph memory leak
niklashoelter
committed
test(autobatching): cover _detach_state_graph grad-drop for the UMA leak fix
niklashoelter
committed

Commits on Jul 5, 2026

fix(autobatching): detach states in _chunked_apply init pass
niklashoelter
committed
refactor(autobatching): promote detach_state_graph to public API
niklashoelter
committed

Commits on Jul 21, 2026