Try bootstrapping tpu env from env vars (#4499)
* Try bootstrapping tpu env from env vars
* Try bootstrapping tpu env from env vars
* Formatting
* Mock env vars
* Use TPU_SKIP_MDS_QUERY
---------
Co-authored-by: root <root@t1v-n-804806aa-w-0.us-central2-b.c.tpu-pytorch.internal>