Awesome, glad to hear that helped. What’s nice is that mixed-precision training it’s so easy to enable but can have a huge impact in practice 🙂

→ View original post on X — @rasbt,