Skip to content

Pull requests: AI-Hypercomputer/maxtext

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

Update HF login in jupyter notebooks
#4868 opened Aug 13, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
2
Escalate infrastructure issue: pod not found
#4867 opened Aug 13, 2026 by olufiyin19 Collaborator Loading…
1
fix(post_train): Fix Tunix is_update_step signature and Qwix LoRA FSDP mesh sharding
#4866 opened Aug 13, 2026 by RexBearIU Collaborator • Draft
2 of 4 tasks
3
Add SFT vision COCO captions dataset and config support gemini-review
#4864 opened Aug 12, 2026 by subawocit Collaborator Loading…
4 tasks done
12
Gemma-4 small (E2B/E4B): emit vision-tower shapes in HF export map
#4860 opened Aug 12, 2026 by lokic233 Loading…
3 of 4 tasks
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
4
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
8
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 Loading…
4 tasks done
27
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
8
Reject KV head counts that cannot be sharded across the tensor-parallel mesh.
#4811 opened Aug 10, 2026 by copybara-service Bot Loading…
1
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
69
Fix multihost_utils.process_allgather tiled=True in hf_utils.py
#4799 opened Aug 9, 2026 by olufiyin19 Collaborator Loading…
7
ProTip! Adding no:label will show everything without a label.