Automatic mixed precision native in Pytorch nightly

I am going to experiment with it this weekend but I had a crazy idea to make this work natively with v2. Stay tuned!

7 Likes