straight-through-gradient topic

List straight-through-gradient repositories

relaxit

26
Stars
1
Forks
26
Watchers

[BMM 24-25] "Just Relax It": Implementation of different relaxation methods

PyTorch-Gumbel-Sigmoid

21
Stars
1
Forks
21
Watchers

Implementation of the Gumbel-Sigmoid distribution in PyTorch.