Adapting Learning Rates in Full Precision Training to Half Precision Training on TPUs michael jemisonMay 2, 2023