Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
239 Open 4,172 Closed
239 Open 4,172 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

Add Pathways health check to run_pathways_tests.yml
#4697 opened Jul 31, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
1
PR permission test, from TPU Raiden team intern, clarify that maxtext…
#4696 opened Jul 31, 2026 by haochengd-google Loading…
1
apple envy moe implementation pull ready
#4695 opened Jul 31, 2026 by entrpn Collaborator Loading…
4 tasks done
5
Update Orbax to 0.12.2 (Pathways Multi-Tiered Checkpointing Support)
#4694 opened Jul 31, 2026 by SujeethJinesh Collaborator • Draft
4 tasks done
2
PR #4640: fix: refine scanned layers unroll logic for Gemma 3
#4693 opened Jul 31, 2026 by copybara-service Bot Loading…
4 tasks done
1
Fix compile_cache_test flakiness caused by the persistent cache compile time threshold
#4692 opened Jul 31, 2026 by bzantium Collaborator Loading…
4 tasks done
2
Validate rampup batch size settings in the Pydantic config
#4691 opened Jul 31, 2026 by Atishyy27 Loading…
1
1
fix(ci): resolve CI failures for PR #4501 sharding and Tunix adapter rollout regression
#4689 opened Jul 31, 2026 by copybara-service Bot Loading…
4 tasks done
1
Add Qwen3-VL-2B E2E test scripts
#4688 opened Jul 31, 2026 by chiajunglien Contributor • Draft
4 tasks
1
Add TPU Ulysses context parallelism gemini-review
#4687 opened Jul 31, 2026 by huytransformer Collaborator • Draft
4 tasks
5
Add GPT-OSS-20b E2E test scripts
#4686 opened Jul 31, 2026 by chiajunglien Contributor • Draft
4 tasks done
1
[NNX] Delete Linen (12.1c): collapse dispatch in quantization and model creation
#4685 opened Jul 31, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
2
[NNX] Delete Linen (12.1b): collapse dispatch in sharding and muon
#4684 opened Jul 31, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
2
[NNX] Delete Linen (12.1a): collapse dispatch in the core train loop and maxtext_utils
#4683 opened Jul 31, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
2
Align MaxTextTrainingEngine microbatch accumulation scaling and primary loss metric recording.
#4682 opened Jul 31, 2026 by copybara-service Bot Loading…
@A9isha
Internal change
#4680 opened Jul 30, 2026 by copybara-service Bot Loading…
@SurbhiJainUSC
1
Support saving and restoring intra-step checkpoints in MaxTextTrainingEngine.
#4679 opened Jul 30, 2026 by copybara-service Bot Loading…
@SurbhiJainUSC
1
Add regular scan support for qwen3-next model
#4677 opened Jul 30, 2026 by NuojCheng Collaborator • Draft
4 tasks
2
feat(ci): allow re-running tests without recreating docker images
#4676 opened Jul 30, 2026 by hsuan-lun-chiang Collaborator • Draft
1
Add launch scripts and configurations for Distillation Training.
#4674 opened Jul 29, 2026 by copybara-service Bot Loading…
1
Logits checker token
#4673 opened Jul 29, 2026 by lydhr Collaborator • Draft
4 tasks
1
Copybara auto squash
#4672 opened Jul 29, 2026 by lydhr Collaborator Loading…
4 tasks done
1
Add images in SFT loss_func for multimodal post training
#4670 opened Jul 29, 2026 by dgouju Loading…
2 tasks
1
Dsv4 attention opt on lb changes
#4669 opened Jul 29, 2026 by dipakg-lang Collaborator • Draft
4 tasks
2
Suppressions for pyrefly
#4668 opened Jul 29, 2026 by copybara-service Bot Loading…
ProTip! Filter pull requests by the default branch with base:main.