-
Notifications
You must be signed in to change notification settings - Fork 594
Pull requests: AI-Hypercomputer/maxtext
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
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
[Maxtext][vLLM] Fix embedding sharding rules in RL config and normalize MRoPE position shapes for vLLM
#5081
opened Aug 31, 2026 by
YixuanWang-99
Collaborator
Loading…
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
Upgrade drjax to 0.2.1
#5075
opened Aug 31, 2026 by
khatwanimohit
Collaborator
Loading…
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…
Support 2D transposed shape alignment in model_creation_utils
#5071
opened Aug 31, 2026 by
suhithr
Collaborator
Loading…
4 tasks done
Migrate scripts and docs from tfds pipeline to grain
pull ready
#5070
opened Aug 31, 2026 by
aireenmei
Collaborator
Loading…
4 tasks done
Fix KL divergence regression for Qwen3 in AUTO shard mode
#5067
opened Aug 31, 2026 by
chiajunglien
Collaborator
Loading…
4 tasks done
Expose enable_continuous_profiling CLI flag in MaxText RL
#5061
opened Aug 29, 2026 by
youchunni
Loading…
3 of 4 tasks
Match Tunix peft_trainer_v2 performance in MaxTextTrainingEngine
#5060
opened Aug 29, 2026 by
NuojCheng
Collaborator
Loading…
4 tasks done
Multimodal evaluation runner and end-to-end pipeline for omni POC
#5059
opened Aug 29, 2026 by
subawocit
Collaborator
Loading…
4 tasks done
MaxText/Tunix: Expose enable_continuous_profiling CLI flag in MaxText and Tunix profiler to allow saving >2GB profile as xplane.riegeli
#5058
opened Aug 28, 2026 by
copybara-service
Bot
Loading…
Add a remat checkpoint after the MoE expert-parallel combine
#5057
opened Aug 28, 2026 by
NuojCheng
Collaborator
Loading…
4 tasks done
test(fp8): add PyTorch reference parity suite and logit checker metrics
#5055
opened Aug 28, 2026 by
snehalv2002
Collaborator
•
Draft
3 of 4 tasks
Previous Next
ProTip!
Type g i on any issue or pull request to go back to the issue listing page.