Since downcasting on TPUs is better for calculating metrics and logging values, it might be nice to have an API that lets you toggle downcasting via a context manager. Under the hood this should use ...
Some results have been hidden because they may be inaccessible to you
Show inaccessible results