Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
234 Open 2,126 Closed
234 Open 2,126 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

Add muon optimizer
#2546 opened Oct 25, 2025 by shuningjin • Draft
4 tasks
Add script to run evaluation before and after SFT training
#2545 opened Oct 24, 2025 by SurbhiJainUSC • Draft
4 tasks done
Integrate Diagon SDK for managed profiler
#2544 opened Oct 24, 2025 by xibinliu Loading…
4 tasks done
4
Refactor: 3.1 phase of RESTRUCTURE.md, focussing on setup.sh and related Python dependency & Dockerfile files
#2541 opened Oct 24, 2025 by SamuelMarks Loading…
4 tasks done
[WIP] Support scale down when loading grain checkpoint
#2537 opened Oct 22, 2025 by aireenmei • Draft
4 tasks done
Add user guide to update MaxText dependencies with seed-env
#2536 opened Oct 22, 2025 by bvandermoon Loading…
4 tasks done
7
Add rampup batch size support in MaxText gemini-review
#2535 opened Oct 22, 2025 by NuojCheng Loading…
4 tasks done
3
Fix: Correct path to extra_deps_from_github.txt
#2533 opened Oct 22, 2025 by AidenYu1673 Loading…
4 tasks done
1
[Do Not Merge] Run Tests against jax 0.8.0 image
#2532 opened Oct 21, 2025 by Rohan-Bierneni • Draft
4 tasks done
GRPO refactoring
#2530 opened Oct 21, 2025 by mydatascience Loading…
4 tasks done
11
Qwen3-Next Gated FullAttention Implementation
#2529 opened Oct 21, 2025 by Rohan-Bierneni Loading…
4 tasks done
26
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
6
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…
4 tasks done
1
@aireenmei
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
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
5
ProTip! Type g i on any issue or pull request to go back to the issue listing page.