diff --git a/pyproject.toml b/pyproject.toml index dbf8d8c68..0e3bd9f5b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -106,7 +106,7 @@ gcp = [ tpu = [ "axlearn[gcp]", "jax[tpu]==0.5.3", # must be >=0.4.19 for compat with v5p. - "pathwaysutils==0.1.1", # For JAX+Pathways single-controller accelerator coordinator. + "pathwaysutils==0.1.2", # For JAX+Pathways single-controller accelerator coordinator. ] # Vertex AI tensorboard. TODO(markblee): Merge with `gcp`. vertexai_tensorboard = [