-
Notifications
You must be signed in to change notification settings - Fork 597
Pull requests: AI-Hypercomputer/maxtext
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Migrate MaxText training engine checkpointing to Orbax v1 Checkpointer API
#5103
opened Sep 2, 2026 by
copybara-service
Bot
Loading…
Fix ImportError by pinning tensorflow and tensorflow-text to 2.20.x
#5102
opened Sep 2, 2026 by
chiajunglien
Collaborator
Loading…
4 tasks done
Fix RL Training - vLLM sampler TypeError
#5101
opened Sep 2, 2026 by
chiajunglien
Collaborator
Loading…
4 tasks done
Add RL sequence-packing and vLLM block-size knobs
#5100
opened Sep 2, 2026 by
gutianyu-google
•
Draft
4 tasks done
Defer the data-parallel gradient all-reduce to update() under gradient accumulation
#5099
opened Sep 2, 2026 by
NuojCheng
Collaborator
Loading…
Test Fused GDN Backward Pass Kernel w/ MaxText
#5098
opened Sep 2, 2026 by
Rohan-Bierneni
Collaborator
•
Draft
4 tasks done
Training engine: implement eval_step API
#5097
opened Sep 2, 2026 by
SurbhiJainUSC
Collaborator
Loading…
4 tasks done
Update pyconfig to get config defaults from types.py instead of base.yml
#5096
opened Sep 1, 2026 by
bvandermoon
Collaborator
Loading…
4 tasks done
Implement continuous stream chunking for pre-tokenized C4 MLPerf dataset
#5095
opened Sep 1, 2026 by
snehalv2002
Collaborator
•
Draft
4 tasks
Fixed e2e test for Qwen3.5-35b-a3b
#5094
opened Sep 1, 2026 by
AusarYao28
Collaborator
Loading…
4 tasks done
[WIP] [Trellis] Weight Conversion that is compatible with Raiden
#5089
opened Sep 1, 2026 by
YixuanWang-99
Collaborator
Loading…
4 tasks
Forward tpu-inference MoE kernel knobs and surface MoE padding
#5087
opened Sep 1, 2026 by
khatwanimohit
Collaborator
Loading…
4 tasks done
Make tensorflow optional in requirements
#5086
opened Sep 1, 2026 by
aireenmei
Collaborator
Loading…
4 tasks done
Add ML Diagnostics scalar backend and options wiring for MaxText RL.
#5085
opened Sep 1, 2026 by
copybara-service
Bot
Loading…
feat(training_engine): support Raiden-FFI weight synchronization and adopt single synchronizer
gemini-review
#5082
opened Aug 31, 2026 by
igorts-git
Collaborator
•
Draft
4 tasks done
Retry with rbf = -1 if tokens are dropped
pull ready
#5080
opened Aug 31, 2026 by
Shuwen-Fang
Collaborator
Loading…
4 tasks
Add StreamIndex Kernel for CSA [Deepseek v4]
#5079
opened Aug 31, 2026 by
octatrifan
Collaborator
•
Draft
2 of 4 tasks
add profiling option to compare_training_engine.py
#5078
opened Aug 31, 2026 by
A9isha
Collaborator
Loading…
4 tasks done
Add layer-by-layer tensor distribution debugging mode in MaxText.
#5077
opened Aug 31, 2026 by
copybara-service
Bot
Loading…
feat(models): add qwen3.5-35b-a3b-fp8 model config and HuggingFace mapping
#5076
opened Aug 31, 2026 by
snehalv2002
Collaborator
•
Draft
4 tasks done
[Qwen3.5] Fix standalone torchax converter for current tpu-inference and wire use_standalone_converter into the rollout
#5073
opened Aug 31, 2026 by
wenxindongwork
Collaborator
•
Draft
Integrate Lineage DeepSeek-V3 58 sparse layer scan into MaxText.
#5072
opened Aug 31, 2026 by
copybara-service
Bot
Loading…
Previous Next
ProTip!
Type g p on any issue or pull request to go back to the pull request listing page.