-
Notifications
You must be signed in to change notification settings - Fork 587
Pull requests: AI-Hypercomputer/maxtext
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[Gemma4] Fix the bug of
use_qk_norm accidently set to False in Gemma 4 configs
#4938
opened Aug 18, 2026 by
Dr-Left
Collaborator
Loading…
4 tasks done
Fix OOM in to_huggingface conversion and stabilize post-training E2E TPU pipelines
#4937
opened Aug 18, 2026 by
SurbhiJainUSC
Collaborator
•
Draft
4 tasks done
Add a sharding-aware implementation of muon to Maxtext.
#4936
opened Aug 18, 2026 by
copybara-service
Bot
Loading…
Add support for Tokamax GMM v2 heuristic tiling function.
#4935
opened Aug 18, 2026 by
copybara-service
Bot
Loading…
Fix duplicate XLA compilation of jit(train_step)
#4934
opened Aug 18, 2026 by
olupton
Contributor
Loading…
4 tasks done
Build the Qwen3-Next shared expert only when shared_experts > 0
#4930
opened Aug 18, 2026 by
WandLZhang
Loading…
Let splash attention exceed 524,288 tokens when packing is off
#4929
opened Aug 18, 2026 by
WandLZhang
Loading…
fix(checkpointing): re-raise fatal checkpointing errors in non-elastic mode
#4927
opened Aug 18, 2026 by
RexBearIU
Collaborator
Loading…
Add HCA Static Compilation for Splash Attention [Deepseek v4]
#4924
opened Aug 17, 2026 by
octatrifan
Collaborator
•
Draft
4 tasks done
cosmos3_nano model bring up (patch PR#4475)
#4920
opened Aug 17, 2026 by
lydhr
Collaborator
Loading…
4 tasks done
Archive older news updates to shorten main list
#4919
opened Aug 17, 2026 by
bvandermoon
Collaborator
Loading…
4 tasks done
Add grain_index_storage_option
pull ready
#4916
opened Aug 17, 2026 by
aireenmei
Collaborator
Loading…
4 tasks done
DSV4: Add indexer_save_selection and indexer_sharded_topk for the CSA indexer
#4915
opened Aug 17, 2026 by
systalyze-ai
Loading…
4 tasks done
DSV4: Add mask hoisting, boolean splash masks, and thresholded indexer membership
#4914
opened Aug 17, 2026 by
systalyze-ai
Loading…
4 tasks done
MoE: remove the replicated-activation repeat in RoutedMoE.permute
#4913
opened Aug 17, 2026 by
systalyze-ai
Loading…
4 tasks done
DSV4: Add mhc_split_axis_contraction to contract the mHC rate and embed axes separately
#4912
opened Aug 17, 2026 by
systalyze-ai
Loading…
4 tasks done
mesh: add mesh_ring_axis to give a mesh axis physical rings on a torus without wraparound
#4911
opened Aug 17, 2026 by
systalyze-ai
Loading…
4 tasks done
MoE: add moe_late_tp_combine so routed experts take the tensor axis as expert parallelism
#4910
opened Aug 17, 2026 by
systalyze-ai
Loading…
4 tasks done
DSV4: Add shard_indexer_acts to shard the CSA indexer scores across heads
#4909
opened Aug 17, 2026 by
systalyze-ai
Loading…
4 tasks done
DSV4: Route compressed and sliding train attention onto the splash kernels
#4908
opened Aug 17, 2026 by
systalyze-ai
Loading…
4 tasks done
Previous Next
ProTip!
Adding no:label will show everything without a label.