Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
182 Open 3,903 Closed
182 Open 3,903 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

fix(ci): Remove hardcoded cloud_runner from maxtext_jupyter_notebooks
#4355 opened Jul 6, 2026 by mesakhcienet Collaborator • Draft
4 tasks done
1
Add video-scaling function for SFT padding gemini-review
#4353 opened Jul 5, 2026 by hengtaoguo Collaborator • Draft
4 tasks done
8
[WIP] deepseek_ocr_2 exp
#4352 opened Jul 5, 2026 by hengtaoguo Collaborator • Draft
4 tasks
1
Use tuple keys in flax flatten/unflatten for checkpoint conversion.
#4351 opened Jul 4, 2026 by copybara-service Bot Loading…
@abhinavclemson
1
Use active elastic devices for Pathways MTC init
#4350 opened Jul 3, 2026 by SujeethJinesh Collaborator Loading…
4 tasks done
1
Fix: resolve config mismatch in vllm_decode.py
#4349 opened Jul 3, 2026 by chiajunglien Loading…
4 tasks done
1
Fuse the Gated Delta Net inter-chunk scan into a Pallas TPU kernel
#4348 opened Jul 3, 2026 by bzantium Collaborator Loading…
4 tasks done
1
2
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
23
[NNX] Checkpoint: persist and restore rngs/dropout across resumes
#4343 opened Jul 2, 2026 by ecnal-cienet Collaborator Loading…
4 tasks done
1
[NNX] Checkpoint restore: error on weights missing from the checkpoint
#4341 opened Jul 2, 2026 by ecnal-cienet Collaborator Loading…
4 tasks done
2
fix(ci): Execute newly added or modified scheduled_only tests in pre-submits gemini-review pull ready
#4339 opened Jul 2, 2026 by mesakhcienet Collaborator Loading…
4 tasks done
15
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
ProTip! Adding no:label will show everything without a label.