GitLab

added return_grad for all types of rnnt loss (#29)

* added return_grad for all types of rnnt loss

* lifted T >= S for regular case

* black reformat

* black -l80 reformat

* fixed s_range adjustment rule