Draft: latent model support
# What does this PR do ?

Adds support to token_level_encoder_decoder module to accept Hidden and Loss transformations and apply them. 

# Usage
* You can add the following to your config : 
hidden_transforms: [MegatronGaussianHiddenTransform()] and it will apply a gaussian transform to the hiddens.