titu1994 / warprnnt_numba

WarpRNNT loss ported in Numba CPU/CUDA for Pytorch

MIT License

16 stars 2 forks source link

Closed titu1994 closed 2 years ago

titu1994 commented 2 years ago

Additions

Add code and tests for alpha and beta kernels for naive RNNT loss implementation with CUDA atomic locks