Skip to content

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

Open
JLiu4Coding wants to merge 9 commits into
RL-Align:mainfrom
JLiu4Coding:codex/gumbel-softmax-sampler
Open

[FEAT][kernels] Fused Gumbel-Softmax sampler — Triton forward, ST hard path, fused backward#215
JLiu4Coding wants to merge 9 commits into
RL-Align:mainfrom
JLiu4Coding:codex/gumbel-softmax-sampler

Support chunked Triton Gumbel-Softmax

622a748
Select commit
Loading
Failed to load commit list.
Sign in for the full log view

Annotations

1 warning
gpu-tests (NVIDIA H100 80GB HBM3, 9.0, 1)
succeeded Jul 23, 2026 in 5m 23s