Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
152 Open 3,620 Closed
152 Open 3,620 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

Dpo integration test - logits comparision against Pytorch
#4026 opened May 30, 2026 by igorts-git Collaborator • Draft
4 tasks done
1
Plumb rl.loss_agg_mode to tunix GrpoConfig
#4025 opened May 30, 2026 by py4 Collaborator Loading…
4 tasks done
4
Maxtwin xm support.
#4024 opened May 29, 2026 by copybara-service Bot Loading…
1
fix: raise RuntimeError when checkpoint step >= config.steps bug Something isn't working gemini-review
#4022 opened May 29, 2026 by Dr-Left Collaborator Loading…
4 tasks done
@Dr-Left @khatwanimohit
6
Set lazy_load_tensors to true by default.
#4021 opened May 29, 2026 by niting Collaborator Loading…
4 tasks done
1
t
#4019 opened May 29, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks
1
Add runner diagnostics and pytest log capture for GPU test debugging
#4018 opened May 29, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks done
1
Add support of Grain input pipeline for DPO.
#4009 opened May 28, 2026 by igorts-git Collaborator • Draft
4 tasks done
1
feat: Add Gemma3 LoRA SFT integration and programmatic weight mapping…
#4004 opened May 28, 2026 by RexBearIU 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
#4002 opened May 28, 2026 by dependabot Bot Loading…
1
Add build_and_test_release_candidate.yml workflow
#3999 opened May 27, 2026 by YixuanWang-99 Collaborator Loading…
4 tasks done
9
Delete legacy DPO implementation gemini-review
#3997 opened May 27, 2026 by igorts-git Collaborator Loading…
4 tasks done
4
Publish MaxText to PyPI when Airflow E2E tests succeed
#3994 opened May 27, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
1
Add Github workflow to trigger MaxText E2E Airflow DAG
#3993 opened May 27, 2026 by SurbhiJainUSC Collaborator • Draft
4 tasks done
1
Update offline distillation and saving top-k teacher logits to be efficient
#3990 opened May 27, 2026 by ajkv-google Collaborator Loading…
4 tasks done
6
Add shard exp on fsdp custom mesh gemini-review
#3988 opened May 27, 2026 by NuojCheng Collaborator Loading…
4 tasks done
10
[NNX] NNX migration prep (7.5/N): finish MaxEngine inference carve-outs gemini-review
#3984 opened May 26, 2026 by ecnal-cienet Collaborator Loading…
4 tasks done
12
Refactor moe.py PR #1 gemini-review
#3981 opened May 26, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks done
@gobbleturk @Shuwen-Fang
24
fix(inference): align vllm_decode with LoRA checkpoint restoration an…
#3978 opened May 26, 2026 by chiajunglien • Draft
4 tasks
6
Test/nnx set defaults true mesa
#3976 opened May 25, 2026 by mesakhcienet Collaborator • Draft
4 tasks
1
[WIP] Consolidate mmutils gemini-review
#3974 opened May 22, 2026 by hengtaoguo Collaborator • Draft
4 tasks
8
docs: QLoRA Documentation and Notebooks
#3970 opened May 22, 2026 by RexBearIU Collaborator Loading…
4 tasks done
15
Extend LoRA for Gemma4
#3969 opened May 22, 2026 by RexBearIU Collaborator Loading…
4 tasks done
7
QLoRA Support + NNX Decoder Sharding Fixes
#3968 opened May 22, 2026 by RexBearIU Collaborator Loading…
4 tasks done
7
Torch-free layer-wise test experiment
#3963 opened May 21, 2026 by hengtaoguo Collaborator • Draft
4 tasks
1
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.