Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
238 Open 2,110 Closed
238 Open 2,110 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

[Do Not Merge] Run Tests against jax 0.8.0 image
#2532 opened Oct 21, 2025 by Rohan-Bierneni • Draft
4 tasks done
Update MaxText requirements
#2531 opened Oct 21, 2025 by khatwanimohit Loading…
4 tasks done
3
GRPO refactoring
#2530 opened Oct 21, 2025 by mydatascience Loading…
4 tasks done
1
Qwen3-Next Gated FullAttention Implementation
#2529 opened Oct 21, 2025 by Rohan-Bierneni Loading…
4 tasks done
test ironwood
#2524 opened Oct 20, 2025 by A9isha • Draft
4 tasks
feat: migrate deepseek batch split to nnx pull ready
#2522 opened Oct 20, 2025 by mesakhcienet Loading…
4 tasks done
5
Fix exception causes in data_loader.py
#2521 opened Oct 18, 2025 by cool-RR Loading…
2
add weight_sum_fp32 config
#2519 opened Oct 17, 2025 by suexu1025 Loading…
4 tasks done
2
New landing page
#2517 opened Oct 17, 2025 by melissawm Loading…
4 tasks done
A trial of apply @fused to optimzier state offloading
#2513 opened Oct 16, 2025 by zhenying-liu • Draft
Introduce gather() and scatter_add() reference functions for token-routing in Mixture-of-Experts models.
#2512 opened Oct 16, 2025 by copybara-service bot Loading…
2
feat(data): Default grain_worker_count to -1 for automatic performance tuning
#2510 opened Oct 16, 2025 by bzantium Loading…
3 of 4 tasks
1
1
Update patch to ensure maxtext images are downgraded to 0.7.0
#2506 opened Oct 15, 2025 by Rohan-Bierneni Loading…
4 tasks done
@Rohan-Bierneni
7
Remat policy + batch size estimation using AOT compilation gemini-review
#2504 opened Oct 15, 2025 by NuojCheng Loading…
4 tasks done
3
Fix JAX version bugs for nightly image
#2503 opened Oct 15, 2025 by hengtaoguo • Draft
4 tasks done
@hengtaoguo @gobbleturk
1
Fix generate_metrics_and_upload_to_big_query
#2502 opened Oct 15, 2025 by ycchenzheng Loading…
4 tasks done
@ycchenzheng
2
Recipe Modularization, DAG Integration, and BigQuery Integration
#2499 opened Oct 15, 2025 by DannyLiCom Loading…
4 tasks done
4
[*] pre-commit run --all-files
#2498 opened Oct 14, 2025 by SamuelMarks Loading…
4 tasks done
11
GRPO & SFT E2E tests draft Draft PR
#2497 opened Oct 14, 2025 by mydatascience Loading…
4 tasks done
1
Splitting GH Actions tests to workers pull ready
#2496 opened Oct 14, 2025 by mydatascience Loading…
4 tasks done
8
Add multimodal documentation
#2493 opened Oct 13, 2025 by hengtaoguo • Draft
4 tasks done
Integrating TE Fused Dense for FP8-DelayedScaling, FP8-CurrentScaling, MXFP8
#2491 opened Oct 13, 2025 by phu0ngng Loading…
3 of 4 tasks
[DO NOT MERGE] Create Test.md
#2489 opened Oct 13, 2025 by aicodecanvas Loading…
4 tasks
1
Adding documentation for optimized model tiers
#2485 opened Oct 10, 2025 by rahul-anand Loading…
4 tasks done
@rahul-anand
2
[WIP]Refactor resource management draft Draft PR
#2480 opened Oct 10, 2025 by mydatascience Loading…
4 tasks done
ProTip! What’s not been updated in a month: updated:<2025-09-21.