Add torch no grad - #17

Open

heagenb03 wants to merge 2 commits into

MASILab:masterfrom

heagenb03:add_torch_no_grad

Commits on Jan 22, 2026

Add intial project files
heagenb03
committed
99a32e6
Add torch.no_grad() to methods with unneccessary gradient calc. - Reduces 50GB of peak memory usage down to ~19GB (64%)
heagenb03
committed
e07d537