diff --git a/dags/inference/maxtext_gpu_inference.py b/dags/inference/maxtext_gpu_inference.py index 695ce7fe0..9867fa9d7 100644 --- a/dags/inference/maxtext_gpu_inference.py +++ b/dags/inference/maxtext_gpu_inference.py @@ -27,7 +27,7 @@ with models.DAG( dag_id="maxtext_gpu_inference", schedule=SCHEDULED_TIME, - tags=["inference_team", "maxtext_gpu", "nightly", "benchmark"], + tags=["inference_team", "maxtext_gpu", "nightly", "benchmark", "maxtext"], start_date=datetime.datetime(2025, 3, 26), catchup=False, ) as dag: diff --git a/dags/multipod/jax_functional_tests.py b/dags/multipod/jax_functional_tests.py index a042709fc..814e3f014 100644 --- a/dags/multipod/jax_functional_tests.py +++ b/dags/multipod/jax_functional_tests.py @@ -27,7 +27,7 @@ with models.DAG( dag_id="jax_functional_tests", schedule=SCHEDULED_TIME, - tags=["multipod_team", "jax", "mlscale_devx"], + tags=["multipod_team", "jax", "mlscale_devx", "maxtext"], start_date=datetime.datetime(2024, 10, 23), catchup=False, ) as dag: diff --git a/dags/multipod/legacy.py b/dags/multipod/legacy.py index fc5ee1ad1..67fd38272 100644 --- a/dags/multipod/legacy.py +++ b/dags/multipod/legacy.py @@ -39,6 +39,7 @@ "stable", "nightly", "mlscale_devx", + "maxtext", ], start_date=datetime.datetime(2024, 1, 10), catchup=False, diff --git a/dags/multipod/mxla_collective_nightly.py b/dags/multipod/mxla_collective_nightly.py index a7facd204..7a17343b1 100644 --- a/dags/multipod/mxla_collective_nightly.py +++ b/dags/multipod/mxla_collective_nightly.py @@ -34,6 +34,7 @@ "mxla_collective", "nightly", "mlscale_perfx", + "maxtext", ], start_date=datetime.datetime(2024, 2, 7), catchup=False,