Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
247 Open 4,129 Closed
247 Open 4,129 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

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
Add E2E TPU testing scripts for GPTOSS-20B
#4641 opened Jul 28, 2026 by chiajunglien Contributor Loading…
4 tasks done
2
fix: refine scanned layers unroll logic for Gemma 3
#4640 opened Jul 28, 2026 by mesakhcienet Collaborator Loading…
4 tasks done
3
Migrate deprecated XLA collective pipelining flags in MaxText
#4639 opened Jul 28, 2026 by copybara-service Bot Loading…
1
MHC Projection Fusion gemini-review
#4638 opened Jul 28, 2026 by muskansh-google Loading…
4 tasks done
6
[DECOUPLED-MODE] checkpoint_conversion: import GCS via gcloud_stub in load_dynamic
#4637 opened Jul 28, 2026 by gulsumgudukbay Collaborator Loading…
6 tasks done
2
[MaxText][RL] Omit Redundant Weight Resharding Prior to Step 0 Evaluation & Add Engram Inheritance
#4636 opened Jul 27, 2026 by igorts-git Collaborator • Draft
4 tasks done
1
ProTip! Follow long discussions with comments:>50.