Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
223 Open 4,071 Closed
223 Open 4,071 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

Migrate single-step SPMD RL trainer from //third_party/py/maxrl/trainer/ to //third_party/py/maxtext/src/maxtext/training_engine/, preserving full file revision history via Piper integration.
#4574 opened Jul 22, 2026 by copybara-service Bot Loading…
@A9isha
[Do not submit] test 5 pull ready
#4573 opened Jul 22, 2026 by aireenmei Collaborator • Draft
4 tasks done
1
fix v7x failure on fractional eval batch test
#4572 opened Jul 22, 2026 by NuojCheng Collaborator • Draft
4 tasks
2
Remove all-to-all operation from TSP/CP in Attention Segmentation Mask
#4571 opened Jul 22, 2026 by dinodeep Collaborator Loading…
4 tasks done
2
Promote TPU nightly docker images to latest without verifying E2E Airflow tests
#4570 opened Jul 22, 2026 by SurbhiJainUSC Collaborator Loading…
4 tasks done
3
Add Gemma 4 tokenizer to assets.
#4569 opened Jul 22, 2026 by copybara-service Bot Loading…
1
Deprecate get_logical_axis_rule_from_config gemini-review
#4568 opened Jul 22, 2026 by NuojCheng Collaborator Loading…
4 tasks done
7
Adapt inspect_checkpoint.py for Automated Ckpt Validation Pipeline (Agent Integration)
#4567 opened Jul 22, 2026 by olufiyin19 Collaborator Loading…
4 tasks done
3
Unify the NNX and Linen restore paths in load_state_if_possible
#4566 opened Jul 22, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
4
Update pathwaysutils version pull ready
#4565 opened Jul 22, 2026 by DannyLiCom Collaborator Loading…
4 tasks done
2
Implement QK attention head chunking in MLA to reduce HBM footprint gemini-review
#4564 opened Jul 22, 2026 by zcjhao Collaborator Loading…
48
Batched Video Encoding and Token Mask Propagation for Qwen3*
#4561 opened Jul 21, 2026 by hengtaoguo Collaborator Loading…
4 tasks done
4
feat: configure standalone checkpoint converters with --pure-nnx flag
#4560 opened Jul 21, 2026 by RexBearIU Collaborator • Draft
4 tasks done
2
Updates to JAX flash attention kernel.
#4559 opened Jul 21, 2026 by copybara-service Bot Loading…
feat(test): add modular E2E TPU testing scripts for GPTOSS-20B
#4556 opened Jul 21, 2026 by RexBearIU Collaborator • Draft
4 tasks done
1
fix(nnx): fix gpt3 paxml checkpoint conversion state map and layout
#4555 opened Jul 21, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks done
1
Adding improved profiles for qwen3 models
#4553 opened Jul 21, 2026 by h1t35h Loading…
4 tasks done
Fix hardcoded grain_worker_count in HF data processing pipelines to prevent OOM while running Multimodal SFT via pathway
#4552 opened Jul 21, 2026 by chiajunglien Contributor Loading…
4 tasks done
2
apple envy moe implementation
#4551 opened Jul 21, 2026 by entrpn Collaborator Loading…
4 tasks done
Migrate the test script to use to_maxtext.py for supported models
#4541 opened Jul 20, 2026 by AusarYao28 Collaborator Loading…
4 tasks done
11
Fix pyconfig access regression in inference_microbenchmark_sweep
#4540 opened Jul 20, 2026 by patemotter Collaborator Loading…
4 tasks done
1
Support load-balanced TPU Tokamax ring attention gemini-review pull ready
#4537 opened Jul 20, 2026 by huytransformer Collaborator Loading…
4 tasks done
6
fix: resolve Gemma 3/4 RL rollout gibberish by unrolling scanned weights for vLLM adapter pull ready
#4536 opened Jul 20, 2026 by mesakhcienet Collaborator Loading…
4 tasks done
5
fix
#4535 opened Jul 20, 2026 by mesakhcienet Collaborator • Draft
4 tasks
2
Fix vocab tiling memory space mismatch under parameter host offload
#4533 opened Jul 20, 2026 by huytransformer Collaborator • Draft
4 tasks
1
ProTip! Adding no:label will show everything without a label.