Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
232 Open 2,158 Closed
232 Open 2,158 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

Onboard explicit sharding to deepseek draft Draft PR
#2579 opened Nov 1, 2025 by NuojCheng • Draft
4 tasks
Update Latest News MaxText installation link to new location
#2577 opened Nov 1, 2025 by bvandermoon Loading…
4 tasks done
Support Grain per_worker_buffer_size gemini-review
#2576 opened Oct 31, 2025 by bernardhan33 • Draft
4 tasks done
3
[src/MaxText/configs/types.py] New pydantic models repesentating all configuration files in MaxText ; [src/MaxText/pyconfig.py] New temporary wrapper to not break existing API ; [src/MaxText/pyconfig_og.py] Move original version here ; [src/MaxText/configs/__init__.py] Make this a module ; [tests/pyconfig_test.py] Import from og pyconfig ; [*requirements*.txt] Add pydantic requirement
#2575 opened Oct 31, 2025 by copybara-service bot Loading…
updated image; added support for adding custom xla/libtpu flags
#2571 opened Oct 31, 2025 by ranuka123 Loading…
2 tasks done
2
Fix generate_metrics_and_upload_to_big_query
#2568 opened Oct 30, 2025 by ycchenzheng Loading…
4 tasks done
1
Refactor Tunix adapter to use a dedicated weight mapping utility.
#2567 opened Oct 30, 2025 by copybara-service bot Loading…
@abhinavclemson
Add gather/scatter based token permute/unpermute to MoE.
#2566 opened Oct 30, 2025 by copybara-service bot Loading…
Add Qwen3-Omni multimodal support
#2565 opened Oct 30, 2025 by eitanporat • Draft
1
Add Qwen3-Omni text-only config gemini-review
#2561 opened Oct 29, 2025 by hengtaoguo Loading…
4 tasks done
@NicoGrande
4
Add a missing input argument to run_test_against_package.yml
#2554 opened Oct 28, 2025 by kanglant Loading…
4 tasks done
[Diloco] add diloco related utils gemini-review
#2552 opened Oct 28, 2025 by khatwanimohit Loading…
4 tasks done
4
fix tokamax default config
#2551 opened Oct 27, 2025 by suexu1025 Loading…
4 tasks done
Refactor Tunix adapter to use model-specific backend mappings.
#2549 opened Oct 27, 2025 by copybara-service bot Loading…
@abhinavclemson
feat: add qwen3-30b-a3b conversion compatibility
#2547 opened Oct 25, 2025 by emergenz Loading…
Add muon optimizer
#2546 opened Oct 25, 2025 by shuningjin • Draft
4 tasks done
6
Integrate SDK for managed profiler
#2544 opened Oct 24, 2025 by xibinliu Loading…
4 tasks done
15
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
46
Add rampup batch size support in MaxText pull ready
#2535 opened Oct 22, 2025 by NuojCheng Loading…
4 tasks done
38
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 gemini-review
#2530 opened Oct 21, 2025 by mydatascience Loading…
4 tasks done
50
Qwen3-Next Gated FullAttention Implementation gemini-review
#2529 opened Oct 21, 2025 by Rohan-Bierneni Loading…
4 tasks done
@Rohan-Bierneni
69
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
11
ProTip! Mix and match filters to narrow down what you’re looking for.