Half or mixed precision?

[jloganolson]

Hello! Are there an examples of doing half or mixed precision? If I'm just using nn.Linear layers, I'm guessing the most straightforward thing to do would be to make a copy where the weights/biases have dtype='float16' tacked on the end (or equivalent using jmp)?