Skip to content

WarpRNNT Numba v0.2.2

Compare
Choose a tag to compare
@titu1994 titu1994 released this 24 Jan 08:25
· 13 commits to master since this release

Initial release of Warp RNNT loss with Numba JIT compile (CPU/CUDA)

Supports:

  1. Pytorch RNNT loss (CPU and JIT compiled CUDA)
  2. FastEmit
  3. Gradient clipping