Feature Request: Context manager for downcasting on TPUs

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 `patch_environment` to patch the right environmental variable.

Proposed API design:

```python
with accelerator.tpu_downcast():
# Run through evaluation loop
```