Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
258 Open 1,942 Closed
258 Open 1,942 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

Better sharding for dsv3 moe layer
#2373 opened Sep 19, 2025 by suexu1025 Loading…
4 tasks done
6
Add workflow for unit tests with Pathways backend
#2371 opened Sep 19, 2025 by wstcliyu Loading…
4 tasks done
1
Migrate LlamaDecoderLayer to NNX
#2370 opened Sep 19, 2025 by RissyRan Loading…
4 tasks done
Remove sharding rules for q_lora and kv_lora from base.yml
#2369 opened Sep 19, 2025 by SurbhiJainUSC Loading…
4 tasks done
@richjames0 @gobbleturk
Testing copybara
#2365 opened Sep 17, 2025 by copybara-service bot Loading…
@gobbleturk
Remove batch truncation for fractional batch sizes.
#2363 opened Sep 17, 2025 by chaserogo Loading…
4 tasks
1
Fix direct uploading of HF checkpoints to GCS.
#2362 opened Sep 17, 2025 by chaserogo Loading…
4 tasks
1
Refactor: third phase of RESTRUCTURE.md
#2361 opened Sep 17, 2025 by SamuelMarks Loading…
4 tasks done
Add automated check for docs build
#2356 opened Sep 16, 2025 by melissawm Loading…
4 tasks done
1
Qwen3 sft collab
#2355 opened Sep 16, 2025 by mydatascience Loading…
4 tasks done
4
feat(input_pipeline): Add support for chunking long sequences instead truncation
#2354 opened Sep 16, 2025 by bzantium Loading…
4 tasks done
1
@aireenmei
13
feat(train): Enable z_loss regularization and logging via z_loss_weight flag
#2353 opened Sep 16, 2025 by bzantium Loading…
4 tasks done
1
[maxtext_jax_ai_image.Dockerfile] MAXTEXT_TEST_ASSETS_DIR → MAXTEXT_TEST_ASSETS_ROOT
#2347 opened Sep 15, 2025 by SamuelMarks Loading…
4 tasks done
Migrate LlamaDecoderLayer to NNX
#2338 opened Sep 12, 2025 by bvandermoon • Draft
4 tasks done
Create docker for pathways with vllm and tpu-commons
#2337 opened Sep 12, 2025 by A9isha • Draft
4 tasks
[WIP] Add more codeowners from the team pull ready
#2326 opened Sep 9, 2025 by RissyRan • Draft
4 tasks done
@shralex @richjames0 @gagika
Logit-Verification for Kimi K2
#2320 opened Sep 9, 2025 by marlinfiggins • Draft
2 of 4 tasks
[Draft] Integrate TE/JAX GroupedGEMM for MoE.
#2319 opened Sep 9, 2025 by mingxu1067 Loading…
4 tasks done
24
[Do not merge] test sentencepiece version
#2316 opened Sep 9, 2025 by aireenmei • Draft
4 tasks done
feat(api_server): Add OpenAI-compatible API server for MaxText models
#2313 opened Sep 8, 2025 by babyplutokurt Loading…
4 tasks done
58
sharding length conflict resolution2
#2312 opened Sep 8, 2025 by shuningjin • Draft
4 tasks
experiment precision
#2308 opened Sep 8, 2025 by hengtaoguo • Draft
4 tasks
Review 'Understand Logs and Metrics' doc
#2306 opened Sep 8, 2025 by melissawm Loading…
4 tasks done
Change attention mask type to uint8 for cudnn_flash_te
#2303 opened Sep 5, 2025 by Skylion007 Loading…
4 tasks done
2
xprof user guide
#2300 opened Sep 5, 2025 by suexu1025 Loading…
4 tasks
8
ProTip! Filter pull requests by the default branch with base:main.