Skip to content

Pull requests: GoogleCloudPlatform/ml-auto-solutions

New pull request New
38 Open 1,104 Closed
38 Open 1,104 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

feat: MaxText Checkpoint Validation Agent Shape and Mock Tensor sub DAGs
#1318 opened Jul 16, 2026 by olufiyin19 Collaborator Loading…
4 tasks done
3
add qwen3.5 35b to xlml tests
#1284 opened Jun 9, 2026 by Rohan-Bierneni Collaborator Loading…
4 tasks done
@Rohan-Bierneni
2
build(deps): Bump apache-airflow from 2.10.2 to 3.2.1rc1 in /.github dependencies Pull requests that update a dependency file python Pull requests that update python code
#1259 opened May 6, 2026 by dependabot Bot Loading…
refactor: Remove maxdiffusion configurations and related code from DAGs
#1221 opened Mar 30, 2026 by michelle-yooh Collaborator Loading…
4 tasks done
Add Qwen3-Next tests to MaxText_MoE DAG
#1194 opened Feb 20, 2026 by Rohan-Bierneni Collaborator Loading…
4 tasks done
@Rohan-Bierneni
10
Migrate gsutil usage to gcloud storage
#1106 opened Dec 24, 2025 by bhandarivijay-png Loading…
Deprecate tpu maxtext candidate image and add JAII_candidate DAG
#782 opened Jul 31, 2025 by Rohan-Bierneni Collaborator Loading…
4 tasks done
change tpu to v5p-8 for SDv2
#760 opened Jun 16, 2025 by notabee Collaborator Loading…
4 tasks done
Upgrade XPK to v0.8.0 and add support for xpk storage command
#747 opened Jun 2, 2025 by raymondzouu Collaborator • Draft
4 tasks
Rishabh/maxdiffusion sdv2 regression
#734 opened May 21, 2025 by notabee Collaborator • Draft
4 tasks
Added extra logging and output to bite TPU unittests
#731 opened May 20, 2025 by asall Collaborator Loading…
4 tasks done
Add regression test for Llama3.1 70B GCSFuse NeMo recipe in main branch
#729 opened May 19, 2025 by lepan-google Collaborator Loading…
2
Added optional custom script post processing task
#719 opened May 12, 2025 by asall Collaborator • Draft
4 tasks done
@asall
Reenabling pathways_maxtext_v5e_configs_perf
#686 opened Apr 22, 2025 by lukebaumann Collaborator • Draft
4 tasks
@RissyRan @abhinavclemson
Maxtext gpu tests on jsts
#644 opened Mar 28, 2025 by Rohan-Bierneni Collaborator • Draft
4 tasks done
Create vllm_xpk.py
#614 opened Mar 5, 2025 by ManfeiBai Collaborator • Draft
4 tasks done
[WIP] Yooh/gpu moe dag a3u
#597 opened Feb 10, 2025 by michelle-yooh Collaborator Loading…
4 tasks
Add team-members to owners
#595 opened Feb 4, 2025 by shralex Collaborator Loading…
4 tasks done
fix torch_xla installation
#584 opened Jan 28, 2025 by zpcore Collaborator Loading…
4 tasks done
[regression test] Update run_mantaray_jobs.py for splited test order for PyTorch regression test on TPU
#572 opened Jan 21, 2025 by ManfeiBai Collaborator Loading…
4 tasks done
6
Split test in prod or dev environment for resource limitation
#551 opened Jan 14, 2025 by ManfeiBai Collaborator Loading…
4 tasks done
5
Remove tpu configs from AOT test and update HybridSim test name
#550 opened Jan 14, 2025 by raymondzouu Collaborator Loading…
4 tasks done
2
[IN PROGRESS] Install full xpk dependency
#494 opened Nov 28, 2024 by RissyRan Collaborator • Draft
4 tasks done
[WIP] Update pytorchxla_torchbench.py
#478 opened Nov 15, 2024 by ManfeiBai Collaborator Loading…
4 tasks
Run the GCE JAX test on nightly, stable and stable stack
#468 opened Nov 8, 2024 by guptaaka Collaborator Loading…
4 tasks done
ProTip! no:milestone will show everything without a milestone.