Is torchaudio.functional.rnnt_loss() a drop-in replacement for torch.nn.functional.ctc_loss() ?
torchaudio.functional.rnnt_loss()
torch.nn.functional.ctc_loss()