Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
224 Open 4,334 Closed
224 Open 4,334 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

CI: Run CI tests on stable deps in Docker image pipelines and explicitly name test jobs
#4848 opened Aug 12, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
2
Gemma4e2b jax.checkpoint & remat fix
#4847 opened Aug 12, 2026 by mattdonati Loading…
4 tasks done
4
Add more elasticity goodput tests
#4846 opened Aug 12, 2026 by dipannita08 Collaborator Loading…
4 tasks done
3
Add max_inflight_computations config parameter and clean up getattr lookups
#4845 opened Aug 11, 2026 by copybara-service Bot Loading…
1
test code style
#4844 opened Aug 11, 2026 by copybara-service Bot Loading…
@shuningjin
Fix muon moe dimension numbers gemini-review
#4843 opened Aug 11, 2026 by parsley9877 Collaborator Loading…
4 tasks done
1
7
add zxhe-sean to CODEOWNERS
#4841 opened Aug 11, 2026 by zxhe-sean Collaborator Loading…
4 tasks done
2
custom processor for omni poc gemini-review
#4839 opened Aug 11, 2026 by subawocit Collaborator Loading…
4 tasks done
10
Add TPU USP context parallelism gemini-review
#4836 opened Aug 11, 2026 by huytransformer Collaborator • Draft
4 tasks
10
Support structured LossOutput and WeightedMetric in MaxText training engine.
#4835 opened Aug 11, 2026 by copybara-service Bot Loading…
@SurbhiJainUSC
1
Add MaxText mHC-lite Pallas/Mosaic TPU kernel and profiling benchmarks.
#4834 opened Aug 11, 2026 by copybara-service Bot Loading…
@dandragona
1
add glm5.2 indexshare
#4832 opened Aug 11, 2026 by notabee Collaborator • 2/2 Loading…
4 tasks done
8
Support activation rematerialization with small Gemma models.
#4830 opened Aug 11, 2026 by copybara-service Bot Loading…
1
Fix installation guide link in build_maxtext.md
#4828 opened Aug 11, 2026 by lydhr Collaborator • Draft
4 tasks
3
Add multi_tier_checkpointing_backup_interval_steps param
#4827 opened Aug 11, 2026 by copybara-service Bot Loading…
Support inserting and replaying router expert decisions in trainer
#4826 opened Aug 11, 2026 by khatwanimohit Collaborator • Draft
4 tasks
5
Support extracting router replay expert indices from MaxText inference
#4813 opened Aug 10, 2026 by khatwanimohit Collaborator Loading…
4 tasks done
3
Quantize activations before EP all gather
#4812 opened Aug 10, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks
9
Reject KV head counts that cannot be sharded across the tensor-parallel mesh.
#4811 opened Aug 10, 2026 by copybara-service Bot Loading…
Gemma-4 vision: correct std_bias/std_scale classification per HF standardize (fidelity fix)
#4807 opened Aug 10, 2026 by lokic233 • Draft
3
Checkpoint Validation Pipeline Operations: Airflow polling integrations, PyConfig overrides, and pipeline lifecycle tooling
#4800 opened Aug 10, 2026 by olufiyin19 Collaborator Loading…
4 tasks done
62
Fix multihost_utils.process_allgather tiled=True in hf_utils.py
#4799 opened Aug 9, 2026 by olufiyin19 Collaborator Loading…
7
Add gdn_matmul_precision opt-in knob for Qwen3-Next Gated DeltaNet
#4795 opened Aug 8, 2026 by Noman654 Loading…
1
Gemma-4 E2B: enable image (multimodal) E2E parity — vision clipped-linears + padded-patch masking + PLE substitution (E4B vision-tower validated; E4B E2E pending)
#4790 opened Aug 8, 2026 by lokic233 Loading…
10
Add JAX compilation cache clear fixture to prevent XLA memory accumulation gemini-review
#4784 opened Aug 7, 2026 by SurbhiJainUSC Collaborator Loading…
4 tasks done
6
ProTip! Filter pull requests by the default branch with base:main.