[FEAT][kernels] Fused Gumbel-Softmax sampler — Triton forward, ST hard path, fused backward - #215

[FEAT][kernels] Fused Gumbel-Softmax sampler — Triton forward, ST hard path, fused backward#215

JLiu4Coding wants to merge 12 commits into

RL-Align:mainRL-Align/RL-Kernel:mainfrom

JLiu4Coding:codex/gumbel-softmax-samplerJLiu4Coding/RL-Kernel:codex/gumbel-softmax-samplerCopy head branch name to clipboard