Skip to content

Reduce peak backward memory in GumbelSoftmaxFunction (in-place grad arithmetic)#7

Open
Nakanokensetsu wants to merge 1 commit into
IST-DASLab:mainfrom
Nakanokensetsu:fix/inplace-gumbel-backward-mem
Open

Reduce peak backward memory in GumbelSoftmaxFunction (in-place grad arithmetic)#7
Nakanokensetsu wants to merge 1 commit into
IST-DASLab:mainfrom
Nakanokensetsu:fix/inplace-gumbel-backward-mem

Commits

Commits on Jul 14, 2026