-
Notifications
You must be signed in to change notification settings - Fork 525
Pull requests: AI-Hypercomputer/maxtext
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[Qwen3.5] Add moe weight sync script
#4041
opened Jun 2, 2026 by
Rohan-Bierneni
Collaborator
Loading…
4 tasks done
Support Qwix quantization on NNX
#4040
opened Jun 2, 2026 by
hsuan-lun-chiang
Collaborator
•
Draft
4 tasks done
update rope_max_timescale to 1M for qwen3-30b-a3b-base to match HF
#4039
opened Jun 2, 2026 by
JamesDeng42
Collaborator
Loading…
4 tasks done
[NNX] NNX migration (12/N): delete Linen code paths, classes, and NNX compatibility flags
#4038
opened Jun 2, 2026 by
ecnal-cienet
Collaborator
•
Draft
4 tasks done
Add MoE router similarity and expert fraction distillation metrics
#4037
opened Jun 1, 2026 by
JamesDeng42
Collaborator
Loading…
4 tasks done
Replace 'with mesh' with 'with jax.set_mesh(mesh)'
pull ready
#4034
opened Jun 1, 2026 by
SurbhiJainUSC
Collaborator
Loading…
4 tasks done
[SC] Specify tiling in mosaic kernel so it does not rely on global flag xla_tpu_use_tc_device_shape_on_sc.
#4033
opened Jun 1, 2026 by
copybara-service
Bot
Loading…
Add dataset_processor_path CLI knob for custom datasets
#4031
opened Jun 1, 2026 by
py4
Collaborator
Loading…
4 tasks done
Add eval_batch_size knob for faster post-train RL evaluation
#4030
opened Jun 1, 2026 by
py4
Collaborator
Loading…
4 tasks done
get_optimizer: respect learning_rate_schedule_steps config knob
#4029
opened Jun 1, 2026 by
py4
Collaborator
Loading…
3 tasks done
Add distillation launchers for qwen3-30b-a3b-base and gpt-oss-20b
gemini-review
#4028
opened May 31, 2026 by
gagika
Collaborator
Loading…
4 tasks done
Qwen3.5 vision layers ckpt conversion and decode
#4027
opened May 31, 2026 by
hengtaoguo
Collaborator
Loading…
4 tasks done
Dpo integration test - logits comparision against Pytorch
#4026
opened May 30, 2026 by
igorts-git
Collaborator
•
Draft
4 tasks done
Plumb rl.loss_agg_mode to tunix GrpoConfig
pull ready
#4025
opened May 30, 2026 by
py4
Collaborator
Loading…
4 tasks done
fix: raise RuntimeError when checkpoint step >= config.steps
bug
Something isn't working
gemini-review
#4022
opened May 29, 2026 by
Dr-Left
Collaborator
Loading…
4 tasks done
Remove LD LIBRARY PATH to fix build image failures
#4019
opened May 29, 2026 by
Shuwen-Fang
Collaborator
Loading…
4 tasks done
Add runner diagnostics and pytest log capture for GPU test debugging
#4018
opened May 29, 2026 by
Shuwen-Fang
Collaborator
Loading…
4 tasks done
Add support of Grain input pipeline for DPO.
#4009
opened May 28, 2026 by
igorts-git
Collaborator
Loading…
4 tasks done
Bump the pip group across 1 directory with 3 updates
dependencies
Pull requests that update a dependency file
python
Pull requests that update python code
#4002
opened May 28, 2026 by
dependabot
Bot
Loading…
Publish MaxText to PyPI when Airflow E2E tests succeed
#3994
opened May 27, 2026 by
SurbhiJainUSC
Collaborator
•
Draft
4 tasks done
Add Github workflow to trigger MaxText E2E Airflow DAG
#3993
opened May 27, 2026 by
SurbhiJainUSC
Collaborator
•
Draft
4 tasks done
Previous Next
ProTip!
Find all pull requests that aren't related to any open issues with -linked:issue.