Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
176 Open 3,901 Closed
176 Open 3,901 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

feat(nnx): support native Flax NNX PEFT/LoRA training loop
#4346 opened Jul 3, 2026 by RexBearIU Collaborator • Draft
4 tasks done
1
Use tuple keys in flax flatten/unflatten for checkpoint conversion.
#4345 opened Jul 3, 2026 by copybara-service Bot Loading…
@abhinavclemson
1
feat(scan_layers): dynamically set scan_layers from checkpoint metadata
#4344 opened Jul 3, 2026 by RexBearIU Collaborator Loading…
4 tasks done
4
[NNX] Checkpoint: persist and restore rngs/dropout across resumes
#4343 opened Jul 2, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
[NNX] Checkpoint restore: error on weights missing from the checkpoint
#4342 opened Jul 2, 2026 by ecnal-cienet Collaborator • Draft
4 tasks done
1
[NNX] Checkpoint restore: name the parameter path on ShapeDtypeStruct mismatch
#4341 opened Jul 2, 2026 by ecnal-cienet Collaborator Loading…
4 tasks done
1
fix(ci): Execute newly added or modified scheduled_only tests in pre-submits
#4339 opened Jul 2, 2026 by mesakhcienet Collaborator Loading…
4 tasks done
14
bug fixes for deepseek4
#4337 opened Jul 2, 2026 by snehalv2002 Collaborator • Draft
4 tasks
1
checkpoint conversion support for deepseek v4
#4336 opened Jul 2, 2026 by snehalv2002 Collaborator • Draft
4 tasks
1
fix grain_worker_count max value
#4335 opened Jul 2, 2026 by aireenmei Collaborator Loading…
4 tasks done
1
Remove stale Tunix degenerate group masking config
#4334 opened Jul 2, 2026 by AntonyMei Collaborator Loading…
4 tasks done
1
1
Move data_processing_test to cpu
#4333 opened Jul 2, 2026 by aireenmei Collaborator • Draft
4 tasks done
1
Automated Code Change
#4332 opened Jul 1, 2026 by copybara-service Bot Loading…
1
Automated Code Change
#4331 opened Jul 1, 2026 by copybara-service Bot Loading…
1
add Qwen3 VL tests, update logit checker, and golden logit generator
#4330 opened Jul 1, 2026 by subawocit Collaborator • Draft
4 tasks done
1
Use synthetic data for scripts under src/maxtext/configs/tpu
#4329 opened Jul 1, 2026 by aireenmei Collaborator • Draft
4 tasks done
1
Attempt to Update to jax 0.10.1
#4328 opened Jul 1, 2026 by gobbleturk Collaborator Loading…
4 tasks
1
Avoid checkpoint work on skipped steps
#4326 opened Jul 1, 2026 by SujeethJinesh Collaborator Loading…
4 tasks done
1
Fix HF token validation to support Hugging Face CLI cache
#4325 opened Jul 1, 2026 by copybara-service Bot Loading…
1
Bump the pip group across 3 directories with 5 updates dependencies Pull requests that update a dependency file python Pull requests that update python code
#4324 opened Jul 1, 2026 by dependabot Bot Loading…
1
Add experimental attention quantization flags.
#4321 opened Jul 1, 2026 by copybara-service Bot Loading…
1
[Do Not Merge] Fixes in forward pass logits for V4
#4319 opened Jul 1, 2026 by Rohan-Bierneni Collaborator • Draft
4 tasks
1
ci: add ci fail-fast implementation plan
#4315 opened Jul 1, 2026 by mesakhcienet Collaborator Loading…
4 tasks done
12
Refactor checkpointing to use Orbax v1.
#4314 opened Jul 1, 2026 by copybara-service Bot Loading…
1
Fix HF token validation to support Hugging Face CLI cache pull ready
#4313 opened Jul 1, 2026 by Shuwen-Fang Collaborator Loading…
4 tasks done
@Shuwen-Fang
3
ProTip! Exclude everything labeled bug with -label:bug.