Add support for setting the GCS credential

Set the credential for both the TPU and the local notebook.

https://b.corp.google.com/issues/158133824