-
Notifications
You must be signed in to change notification settings - Fork 601
Pull requests: AI-Hypercomputer/maxtext
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Trace the engine's eval kernel under the mesh and the axis rules
#5144
opened Sep 4, 2026 by
NuojCheng
Collaborator
Loading…
4 tasks
Clip gradients with the norm the engine already computed
#5143
opened Sep 4, 2026 by
NuojCheng
Collaborator
Loading…
4 tasks done
Add separate final normalization for Multi-Token Prediction in DeepSeek models.
#5142
opened Sep 4, 2026 by
copybara-service
Bot
Loading…
Enable quantized multi-token-prediction
#5141
opened Sep 4, 2026 by
shuningjin
Collaborator
Loading…
4 tasks done
vLLM adapter: index kv_caches through layer_name_to_kvcache_index
#5139
opened Sep 4, 2026 by
sierraisland
Contributor
Loading…
4 tasks done
Fix AttributeError on missing use_sft in grain data processing
#5137
opened Sep 4, 2026 by
bvandermoon
Collaborator
Loading…
4 tasks done
Keep qwix out of tpu-inference's fused MoE kernel and pre-quantize weights
#5136
opened Sep 3, 2026 by
sierraisland
Contributor
Loading…
4 tasks done
fix(attention test): give the mask tests the axis rules the cp gate reads
#5134
opened Sep 3, 2026 by
gulsumgudukbay
Collaborator
Loading…
4 tasks done
fix(mhc test): let the batch scale with the device count
#5133
opened Sep 3, 2026 by
gulsumgudukbay
Collaborator
Loading…
4 tasks done
fix(diloco test): set enable_checkpointing explicitly when loading state
#5132
opened Sep 3, 2026 by
gulsumgudukbay
Collaborator
Loading…
4 tasks done
fix(nnx): keep only newly created params in _apply_layers_sequentially
#5130
opened Sep 3, 2026 by
hodaaaaaaaaaa
Loading…
3 of 4 tasks
feat(kda): integrate KDA attention with tokamax backend and CP support
#5128
opened Sep 3, 2026 by
chiaotung97
Collaborator
Loading…
4 tasks done
Add SparseCore collective offloading for MoE with custom VJP for FSDP All-Gather to Reduce-Scatter.
#5125
opened Sep 3, 2026 by
copybara-service
Bot
Loading…
[megablox] Remove redundant expert-weight transpose copy in unquantized gmm_v2 DLHS
#5123
opened Sep 3, 2026 by
copybara-service
Bot
Loading…
Add flag governing whether to apply Muon to MoE routers.
#5122
opened Sep 3, 2026 by
copybara-service
Bot
Loading…
fix(dependencies): reduce container image vulnerabilities in TPU stable images
pull ready
#5121
opened Sep 3, 2026 by
khatwanimohit
Collaborator
Loading…
4 tasks done
Onboard Qwen3.5 and Qwen3-Next to explicit sharding and ZeRO-1 gradient accumulation
#5119
opened Sep 3, 2026 by
NuojCheng
Collaborator
Loading…
4 tasks done
Support MLPerf training logging compliance for MaxText
gemini-review
#5118
opened Sep 3, 2026 by
zxhe-sean
Collaborator
Loading…
4 tasks done
Implement QK attention head chunking for CSA [Deepseek v4]
#5116
opened Sep 3, 2026 by
octatrifan
Collaborator
•
2/2
Loading…
4 tasks done
Previous Next
ProTip!
Find all pull requests that aren't related to any open issues with -linked:issue.