[FEAT][kernels] Fused Gumbel-Softmax sampler — Triton forward, ST hard path, fused backward - #215
Open
JLiu4Coding wants to merge 10 commits into
Open
[FEAT][kernels] Fused Gumbel-Softmax sampler — Triton forward, ST hard path, fused backward#215JLiu4Coding wants to merge 10 commits into
JLiu4Coding wants to merge 10 commits into
