Add support for downcasting bf16 on TPUs - #523

Merged

muellerzr merged 7 commits into

mainfrom

Jul 20, 2022

Commits on Jul 15, 2022

Allow for downcast
muellerzr
committed
0cd3b0a

Commits on Jul 19, 2022

Refactor to be config only
muellerzr
committed
727dec5
Fin
muellerzr
committed
af9c83b
Fix config default
muellerzr
committed
473cc5f
MOve to cluster config
muellerzr
committed
c609580
Add to state
muellerzr
committed
6759b25
Use getattr
muellerzr
committed
8a4c3fe