Mix-Precision training

[lovodkin93]

Hey, Are there any plans to add support for mixed precision training? I did see in #12 a temporary solution was suggested, but it still throws multiple exceptions relating to mathematical operations between fp16 and fp32 values. Thanks! @rajcscw