Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
175 Open 3,231 Closed
175 Open 3,231 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 config correcting rule and log_config
#3645 opened Apr 11, 2026 by NuojCheng Collaborator Loading…
4 tasks done
2
temp
#3644 opened Apr 11, 2026 by copybara-service bot Loading…
@gagika
1
Add back is_batch_shard_by_expert
#3643 opened Apr 10, 2026 by NuojCheng Collaborator Loading…
4 tasks done
2
Add GEPA prompt optimization tutorial and clean up Qwen3 configs
#3642 opened Apr 10, 2026 by darisoy Collaborator Loading…
4 tasks done
1
Improve environment variable setup in MaxText docs. gemini-review
#3641 opened Apr 10, 2026 by igorts-git Collaborator Loading…
4 tasks done
11
change link of JAX AI Libraries to Cloud docs
#3640 opened Apr 10, 2026 by jacoguzo Collaborator Loading…
4 tasks done
Refactor sharding dump to support custom mesh rule and diverse sharding alternatives
#3639 opened Apr 10, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
Offload gating gmm outputs to host instead of attention output.
#3638 opened Apr 10, 2026 by copybara-service bot Loading…
1
Apply optimization barrier for MLA output to a single microbatch only.
#3637 opened Apr 10, 2026 by copybara-service bot Loading…
1
Integrate gather reduce mosaic SC kernel to deepseekv3 batchsplit.
#3636 opened Apr 10, 2026 by copybara-service bot Loading…
1
Kimi-K2 release update
#3632 opened Apr 10, 2026 by snehalv2002 Collaborator Loading…
4 tasks done
3
Implement batch-split schedule in deepseek_batchsplit.
#3630 opened Apr 10, 2026 by copybara-service bot Loading…
1
Rl debug
#3629 opened Apr 10, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks
1
Fix splash kernel error using CP and explicit sharding
#3628 opened Apr 9, 2026 by NuojCheng Collaborator Loading…
4 tasks done
1
Adding support for fused_moe_gmm
#3627 opened Apr 9, 2026 by NicoGrande Collaborator Loading…
4 tasks
11
Add support for L2 loss in feature distillation.
#3620 opened Apr 9, 2026 by copybara-service bot Loading…
@gagika
1
Add assertion for jax.debug.callback in spike test.
#3618 opened Apr 9, 2026 by copybara-service bot Loading…
@abhinavclemson
1
Remove post-training MaxText shims
#3617 opened Apr 9, 2026 by bvandermoon Collaborator Loading…
4 tasks done
1
Clean up leftover src/MaxText references
#3615 opened Apr 9, 2026 by bvandermoon Collaborator Loading…
4 tasks done
2
Custom Qwen 30B-A3B
#3613 opened Apr 8, 2026 by copybara-service bot Loading…
@gagika
1
Add use_agentic_rollout to test config.
#3611 opened Apr 8, 2026 by copybara-service bot Loading…
@abhinavclemson
1
PR #3427: NNX migration prep (1/N): pure_nnx flag and init_state_fn scaffolding
#3609 opened Apr 8, 2026 by copybara-service bot Loading…
4 tasks done
1
Remove no exp usage from logical rule Part IV draft Draft PR
#3607 opened Apr 8, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
Bump the pip group across 1 directory with 3 updates dependencies Pull requests that update a dependency file python Pull requests that update python code
#3602 opened Apr 8, 2026 by dependabot bot Loading…
1
docs: add LoRA fine-tuning tutorial
#3601 opened Apr 8, 2026 by chiajunglien • Draft
4 tasks
17
ProTip! Mix and match filters to narrow down what you’re looking for.