* Various fixes to support torch script. * Add tests to ensure that the model is torch scriptable. * Update tests.
* Using k2 pruned version transducer loss to train model * Fix style * Minor fixes