Skip to main content

Module ctc

burn_cubecl::kernel

Module ctc 

Source
Expand description

CTC loss kernel

Functionsยง

ctc_alpha_beta
Host entry point for the fused alpha + beta + nll kernel.
ctc_loss
Fused CTC loss for burn-cubecl. Single kernel launch covers the entire alpha recursion across all timesteps.