Any plan on supporting half precision / mixed precision?

[HaoyiZhu]

Hi, thank for the wonderful repo. Do you have any plan to support half or mixed precision training?

I think this is very useful when using high dimensional data and large models as it can greatly save memory.

I noticed that others have asked similar issues before but no reply yet.

Thanks!