WarpRNNT loss ported in Numba CPU/CUDA for Pytorch - View it on GitHub