Skip to content

Flash RNN-T Loss: 2x Faster Loss and up to 50x VRAM Reduction - #16040

Open
MahmoudAshraf97 wants to merge 38 commits into
NVIDIA-NeMo:mainfrom
MahmoudAshraf97:flash-rnnt
Open

Flash RNN-T Loss: 2x Faster Loss and up to 50x VRAM Reduction#16040
MahmoudAshraf97 wants to merge 38 commits into
NVIDIA-NeMo:mainfrom
MahmoudAshraf97:flash-rnnt