All pull requests

Pull requests

• Add Trellis Multi-Host Distributed RL E2E DAG (trellis_multi_host_rl_e2e)

  #1392·dingying-ai opened Oct 2, 2026·4 tasks done·

  ·

• fix(moe): parallelize stable/nightly multicluster chains, shift schedule, and add retries

  #1388·Rohan-Bierneni opened Sep 25, 2026Collaborator·4 tasks done·

  ·

• feat(airflow): Add semantic inference verification for MaxText E2E DAGs

  #1364·chiajunglien opened Sep 8, 2026Contributor·4 tasks done·

  ·

• feat: MaxText Checkpoint Validation Agent Shape and Mock Tensor sub DAGs

  #1318·olufiyin19 opened Jul 16, 2026Collaborator·4 tasks done·

  ·
  3

• add qwen3.5 35b to xlml tests

  #1284·Rohan-Bierneni opened Jun 9, 2026Collaborator·4 tasks done·

  ·
  2

  

• build(deps): Bump apache-airflow from 2.10.2 to 3.2.1rc1 in /.github

  dependenciespython
  #1259·dependabot[bot] opened May 6, 2026Bot·

  ·

• refactor: Remove maxdiffusion configurations and related code from DAGs

  #1221·michelle-yooh opened Mar 30, 2026Collaborator·4 tasks done·

  ·

• Add Qwen3-Next tests to MaxText_MoE DAG

  #1194·Rohan-Bierneni opened Feb 20, 2026Collaborator·4 tasks done·

  ·
  10

  

• Migrate gsutil usage to gcloud storage

  #1106·bhandarivijay-png opened Dec 24, 2025·

  ·

• Deprecate tpu maxtext candidate image and add JAII_candidate DAG

  #782·Rohan-Bierneni opened Jul 31, 2025Collaborator·4 tasks done·

  ·

• change tpu to v5p-8 for SDv2

  #760·notabee opened Jun 16, 2025Collaborator·4 tasks done·

  ·

• Upgrade XPK to v0.8.0 and add support for xpk storage command

  #747·raymondzouu opened Jun 2, 2025Collaborator·4 tasks·

  ·

• Rishabh/maxdiffusion sdv2 regression

  #734·notabee opened May 21, 2025Collaborator·4 tasks·

  ·

• Added extra logging and output to bite TPU unittests

  #731·asall opened May 20, 2025Collaborator·4 tasks done·

  ·

• Add regression test for Llama3.1 70B GCSFuse NeMo recipe in main branch

  #729·lepan-google opened May 19, 2025Collaborator·

  ·
  2

• Added optional custom script post processing task

  #719·asall opened May 12, 2025Collaborator·4 tasks done·

  ·
  

• Reenabling pathways_maxtext_v5e_configs_perf

  #686·lukebaumann opened Apr 22, 2025Collaborator·4 tasks·

  ·
  

• Maxtext gpu tests on jsts

  #644·Rohan-Bierneni opened Mar 28, 2025Collaborator·4 tasks done·

  ·

• Create vllm_xpk.py

  #614·ManfeiBai opened Mar 5, 2025Collaborator·4 tasks done·

  ·

• [WIP] Yooh/gpu moe dag a3u

  #597·michelle-yooh opened Feb 10, 2025Collaborator·4 tasks·

  ·

• Add team-members to owners

  #595·shralex opened Feb 4, 2025Collaborator·4 tasks done·

  ·

• fix torch_xla installation

  #584·zpcore opened Jan 28, 2025Collaborator·4 tasks done·

  ·

• [regression test] Update run_mantaray_jobs.py for splited test order for PyTorch regression test on TPU

  #572·ManfeiBai opened Jan 21, 2025Collaborator·4 tasks done·

  ·
  6

• Split test in prod or dev environment for resource limitation

  #551·ManfeiBai opened Jan 14, 2025Collaborator·4 tasks done·

  ·
  5

• Remove tpu configs from AOT test and update HybridSim test name

  #550·raymondzouu opened Jan 14, 2025Collaborator·4 tasks done·

  ·
  2