It seems like this module is not trained with mixed precision.
Do you plan on adding this feature?
Thank you,
fbcotter
