Add qwen3-next to MaxText_moe DAG

Rohan-Bierneni · Rohan-Bierneni · commit d1d57ef0156a · 2026-02-20T19:48:21.000Z
diff --git a/dags/sparsity_diffusion_devx/maxtext_moe_tpu_e2e.py b/dags/sparsity_diffusion_devx/maxtext_moe_tpu_e2e.py
@@ -73,6 +73,11 @@
           "cluster": XpkClusters.TPU_V5P_8_CLUSTER,
           "time_out_in_min": 90,
       },
+      "qwen3-next-80b": {
+          "script_name": "tpu/qwen/next/qwen3-next-80b-a3b/2_test_qwen3_next_80b_a3b",
+          "cluster": XpkClusters.TPU_V5P_128_CLUSTER,
+          "time_out_in_min": 90, 
+      },
   }
 
   unchained_tests = []
