Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
251 Open 4,130 Closed
251 Open 4,130 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

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/AI-Hypercomputer/maxtext/pull/4513 from Shuwen-Fang:sort_one_sparsecore cccf4ecd7360fbb63557537ccc731eb4f5d0b102
#4666 opened Jul 29, 2026 by copybara-service Bot Loading…
1
Upgrade minimum version of google-cloud-mldiagnostics to 1.0.6 across requirements files.
#4665 opened Jul 29, 2026 by copybara-service Bot Loading…
4 tasks done
1
Mhc optimization: Rewrites einsums for better xla fusion
#4664 opened Jul 29, 2026 by h1t35h Loading…
4 tasks done
1
Upgrade google-cloud-mldiagnostics version
#4663 opened Jul 29, 2026 by madhu-yagadasu Loading…
2
Remove Gemma 3-4B multimodal redundant variable and optimize vllm_decode hbm_utilization of Lama3.1-70b to prevent OOM
#4662 opened Jul 29, 2026 by chiajunglien Contributor Loading…
4 tasks done
2
internal
#4661 opened Jul 29, 2026 by copybara-service Bot Loading…
@abhinavclemson
1
Fix Tunix state translation to handle Flax NNX rng count structures
#4660 opened Jul 29, 2026 by chiajunglien Contributor Loading…
4 tasks done
2
ci: track test flavor in per-test baseline and rename job to Track Test Duration
#4659 opened Jul 29, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks done
2
[DECOUPLED-MODE] ci: run the decoupled test gate in a real decoupled environment
#4658 opened Jul 29, 2026 by gulsumgudukbay Collaborator Loading…
4 tasks done
2
FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/AI-Hypercomputer/maxtext/pull/4513 from Shuwen-Fang:sort_one_sparsecore cccf4ecd7360fbb63557537ccc731eb4f5d0b102
#4657 opened Jul 29, 2026 by copybara-service Bot Loading…
@abhinavclemson
1
FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/AI-Hypercomputer/maxtext/pull/4513 from Shuwen-Fang:sort_one_sparsecore cccf4ecd7360fbb63557537ccc731eb4f5d0b102
#4656 opened Jul 29, 2026 by copybara-service Bot Loading…
@abhinavclemson
1
Conversion Utility Optimization
#4655 opened Jul 29, 2026 by snehalv2002 Collaborator • Draft
4 tasks
1
Warn when running on the Linen code path
#4654 opened Jul 28, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
2
Feat/nnx del linen 1b core utils
#4653 opened Jul 28, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
2
[NNX] Delete Linen (1a/5): collapse dispatch in pre-train checkpoint …
#4652 opened Jul 28, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
2
Fix Zizmor security audit findings in MaxText CI workflow files
#4651 opened Jul 28, 2026 by copybara-service Bot Loading…
@abhinavclemson
1
Exclude optimizer state when running to_huggingface
#4650 opened Jul 28, 2026 by hengtaoguo Collaborator Loading…
4 tasks done
3
Add optimized Pallas kernel for mHC gemini-review
#4649 opened Jul 28, 2026 by dandragona Collaborator • Draft
4 tasks done
2
Extend maxSessionTurns for PR review and assistant gemini-review pull ready
#4648 opened Jul 28, 2026 by RissyRan Collaborator Loading…
4 tasks done
4
fp8 KV cache: pre-dispatch quantize_kv in forward_serve_vllm
#4646 opened Jul 28, 2026 by lokic233 Loading…
2
apple envy moe implementation gemini-review
#4645 opened Jul 28, 2026 by entrpn Collaborator Loading…
4 tasks done
9
Add Simple-evals supports to maxtext runner gemini-review
#4644 opened Jul 28, 2026 by JamesDeng42 Collaborator Loading…
4 tasks done
6
Cache compiled JAX init HLOs in MaxEngine.
#4643 opened Jul 28, 2026 by copybara-service Bot Loading…
1
Run selected tests in Decoupled mode on CPU in CI
#4642 opened Jul 28, 2026 by shralex Collaborator Loading…
4 tasks done
13
fix: refine scanned layers unroll logic for Gemma 3
#4640 opened Jul 28, 2026 by mesakhcienet Collaborator Loading…
4 tasks done
3
ProTip! Follow long discussions with comments:>50.