update bucket name

jax_tpu_embedding/pip_package/cloudbuild.yaml

@@ -26,5 +26,5 @@ options:
 timeout: 5400s
 artifacts:
   objects:
-    location: 'gs://jax-bc-cloud-tpu/wheels/$(date -u +%Y%m%d)'
+    location: 'gs://jax-tpu-embeddings/wheels/$(date -u +%Y%m%d)'
     paths: ['/**/*.whl', '/**/*.txt']

jax_tpu_embedding/pip_package/presubmit.Dockerfile

@@ -1,5 +1,5 @@
 ARG image_name
-ARG base_image="gcr.io/jax-bc-cloud-tpu/${image_name}:latest"
+ARG base_image="gcr.io/jax-tpu-embeddings/${image_name}:latest"
 FROM $base_image
 RUN rm -rf /jax_tpu_embedding/jax_tpu_embedding
 COPY . /jax_tpu_embedding_new