Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
231 Open 2,277 Closed
231 Open 2,277 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

Use threadpool when mixing multiple datasets in Grain
#2703 opened Nov 17, 2025 by aireenmei Loading…
4 tasks done
Introduce Multislice RL
#2702 opened Nov 17, 2025 by xuefgu Loading…
4 tasks done
adding perf members to the owners
#2700 opened Nov 16, 2025 by notabee Loading…
4 tasks done
Change default use_standalone_mappings to True in TunixAdapter.
#2699 opened Nov 16, 2025 by copybara-service bot Loading…
@abhinavclemson
Add Qwen3 Omni Vision Encoder
#2698 opened Nov 16, 2025 by hengtaoguo Loading…
4 tasks done
Support updating grain data mixture during training
#2697 opened Nov 15, 2025 by aireenmei • Draft
4 tasks done
fix tp+tokamax_gmm draft Draft PR
#2696 opened Nov 14, 2025 by NuojCheng • Draft
4 tasks
Use venv and uv throughout in lieu of system Python env + pip
#2695 opened Nov 14, 2025 by SamuelMarks • Draft
4 tasks done
[tools/setup/setup.sh] Nightly cleanup
#2694 opened Nov 14, 2025 by SamuelMarks Loading…
4 tasks done
13
Add GSPO and small fixes draft Draft PR
#2693 opened Nov 14, 2025 by mydatascience Loading…
4 tasks done
quantized ragged dot maxtext integration
#2691 opened Nov 14, 2025 by copybara-service bot Loading…
benchmark some simple layers with some fixes
#2689 opened Nov 14, 2025 by copybara-service bot • Draft
@gobbleturk
[WIP] Bump dependency version
#2686 opened Nov 14, 2025 by RissyRan Loading…
4 tasks done
3
add scheduler config
#2685 opened Nov 14, 2025 by suexu1025 Loading…
4 tasks done
2
Add audio encoder support
#2683 opened Nov 13, 2025 by eitanporat Loading…
4 tasks done
Add checkpoint conversion instructions to GRPO docs gemini-review
#2681 opened Nov 13, 2025 by khatwanimohit Loading…
4 tasks done
12
Add checkpoint conversion parameter mapping for Qwen3 Omni
#2678 opened Nov 13, 2025 by eitanporat Loading…
4 tasks done
[DECOUPLED-MODE] Adding necessary files gemini-review
#2673 opened Nov 12, 2025 by gulsumgudukbay Loading…
4 tasks done
16
Add conversion script for Qwen3 Next and Readme gemini-review
#2672 opened Nov 12, 2025 by Rohan-Bierneni Loading…
4 tasks done
@Rohan-Bierneni
83
[WIP] quantization with tokamax ragged_dot pull ready
#2671 opened Nov 12, 2025 by khatwanimohit • Draft
4 tasks
9
Async Data Loading & Accurate Step Timing to Exclude Profiler Overhead draft Draft PR gemini-review
#2668 opened Nov 12, 2025 by NuojCheng • Draft
3 of 4 tasks
3
Qwen3 deepstack [WIP]
#2660 opened Nov 11, 2025 by eitanporat Loading…
4 tasks
Add MRoPE support for Qwen3-Omni [WIP]
#2659 opened Nov 11, 2025 by eitanporat Loading…
4 tasks
feat: migrate deepseek models to nnx
#2658 opened Nov 11, 2025 by mesakhcienet Loading…
4 tasks done
Fix deepseek tp sharding error draft Draft PR
#2657 opened Nov 11, 2025 by NuojCheng • Draft
4 tasks
ProTip! Mix and match filters to narrow down what you’re looking for.