Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
216 Open 3,962 Closed
216 Open 3,962 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

re-normalize probability distribution when clip_logits_epsilon is used.
#4453 opened Jul 13, 2026 by entrpn Collaborator Loading…
4 tasks done
2
Add multi-directory checkpoint loader for multimodal alignment poc
#4452 opened Jul 13, 2026 by subawocit Collaborator • Draft
4 tasks done
1
[draft] gmm v2 quantization
#4451 opened Jul 13, 2026 by shuningjin Collaborator • Draft
4 tasks
Add igorts-git to all the dirs in code owners. Remove NicoGrande.
#4450 opened Jul 13, 2026 by igorts-git Collaborator Loading…
4 tasks done
1
Abhinav/elastic snapshot
#4449 opened Jul 13, 2026 by abhinavclemson Collaborator Loading…
4 tasks
Keep fused MoE weights intact through FSDP gather (1.5% perf improvement on DeepSeek-v3)
#4448 opened Jul 13, 2026 by abhinavgoel95 Contributor Loading…
4 tasks done
5
implements phase 1 of zero gen verification
#4447 opened Jul 13, 2026 by entrpn Collaborator Loading…
4 tasks done
Fix code quality issue at github head
#4446 opened Jul 13, 2026 by NuojCheng Collaborator Loading…
4 tasks done
1
fix(test): ignore the serialized Mosaic kernel when diffing HLO
#4445 opened Jul 13, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
[Deepseek V4] Decoding Support & Integration with MaxEngine
#4444 opened Jul 13, 2026 by Rohan-Bierneni Collaborator Loading…
4 tasks done
@Rohan-Bierneni
Fix(test): match the current scan_layers mismatch error in test_scan_layers_mismatch_tpu
#4443 opened Jul 13, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
Fix GQA computation when fused_qkv=True
#4442 opened Jul 13, 2026 by dinodeep Collaborator Loading…
4 tasks done
3
Fix pytype late-directive warnings.
#4441 opened Jul 13, 2026 by copybara-service Bot Loading…
fix: relax tolerance in test_flash_attention_block_masked_soft_cap
#4439 opened Jul 13, 2026 by hsuan-lun-chiang Collaborator • Draft
fix(testing): resolve tokenizer loading error and UnboundLocalError in logit checker
#4438 opened Jul 13, 2026 by chiajunglien Contributor Loading…
4 tasks done
8
Optimize MLA generate_mask with TPU-native vectorized threshold cutof…
#4437 opened Jul 13, 2026 by JHCuc3m Collaborator • Draft
1
Abhinavsing/elastic
#4434 opened Jul 11, 2026 by abhinavclemson Collaborator Loading…
1
[Do not submit] test4 pull ready
#4433 opened Jul 11, 2026 by aireenmei Collaborator • Draft
4 tasks
1
Separate RL configurations into a dedicated config class
#4431 opened Jul 10, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
update tpu-inference and vllm versions
#4430 opened Jul 10, 2026 by A9isha Collaborator Loading…
4 tasks done
1
test: Verify Point 3 PR trigger with 1 added test (b/530220873)
#4429 opened Jul 10, 2026 by darisoy Collaborator • Draft
1
Add tutorial for RL GRPO gemma4-e4b on v6e 4x8
#4427 opened Jul 10, 2026 by dasoto Collaborator • Draft
4 tasks done
1
add XLML tests for DeepSeek V4 Flash
#4426 opened Jul 10, 2026 by snehalv2002 Collaborator • Draft
4 tasks
2
Run newly added tests in PRs on TPU7X runners
#4422 opened Jul 10, 2026 by darisoy Collaborator • Draft
4 tasks done
1
Fix text-only batches for multimodal training
#4420 opened Jul 10, 2026 by ybochkov Loading…
1
ProTip! no:milestone will show everything without a milestone.