Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
243 Open 2,064 Closed
243 Open 2,064 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 support for Qwen3-Next Model Architecture
#2487 opened Oct 10, 2025 by parambole Loading…
4 tasks done
@richjames0 @rdyro
@RissyRan @shuningjin @gobbleturk @Rohan-Bierneni
2
fix xlml logits test in gemma, llama4
#2486 opened Oct 10, 2025 by aireenmei Loading…
4 tasks done
4
Adding documentation for optimized model tiers
#2485 opened Oct 10, 2025 by rahul-anand Loading…
4 tasks done
@rahul-anand
1
[GRPO] create a new resharding API which uses split_my_mesh_axis from pathwaysutils
#2482 opened Oct 10, 2025 by khatwanimohit • Draft
4 tasks done
[WIP]Refactor resource management draft Draft PR
#2480 opened Oct 10, 2025 by mydatascience Loading…
4 tasks done
init KVCache fix for grpo trainer gemini-review
#2479 opened Oct 9, 2025 by khatwanimohit Loading…
4 tasks done
4
Fix create_nnx_model() to load NNX checkpoint, verified with vllm_decode gemini-review
#2478 opened Oct 9, 2025 by SurbhiJainUSC Loading…
4 tasks done
6
[TokamaxQwixIntegration] Testing new tokamax splash kernel
#2475 opened Oct 8, 2025 by copybara-service bot Loading…
@jshin1394
[WIP] Add tokamax megablox kernel
#2474 opened Oct 8, 2025 by khatwanimohit • Draft
4 tasks
Adding multi-image support for Llama4 SFT and decode.
#2471 opened Oct 8, 2025 by NicoGrande • Draft
4 tasks done
Onboarding explicit sharding to MaxText gemini-review
#2470 opened Oct 8, 2025 by NuojCheng Loading…
4 tasks done
14
[no merge, draft] Using explicit sharding for zero1
#2465 opened Oct 7, 2025 by NuojCheng • Draft
4 tasks
1
Use ruff over pyink
#2464 opened Oct 7, 2025 by SamuelMarks Loading…
4 tasks done
Create maxtext gpu package and unit/integration tests
#2463 opened Oct 7, 2025 by kanglant Loading…
4 tasks done
Added hooks to RL
#2461 opened Oct 7, 2025 by mydatascience Loading…
4 tasks done
1
Migrate multi_token_prediction to NNX
#2457 opened Oct 6, 2025 by ecnal-cienet Loading…
4 tasks done
2
[exp] nnx copy
#2456 opened Oct 6, 2025 by hengtaoguo • Draft
4 tasks
Initialize pathwaysutils in sft_trainer to run SFT with pathways
#2455 opened Oct 6, 2025 by SurbhiJainUSC Loading…
4 tasks done
Update _input_pipeline_utils.py
#2452 opened Oct 3, 2025 by dasoto Loading…
4 tasks done
Refactor pw_long_running_recipe and merge with mcjax_long_running_recipe
#2449 opened Oct 3, 2025 by ycchenzheng Loading…
4 tasks done
@ycchenzheng
3
Add Qwen3 vLLM weight mapping.
#2444 opened Oct 2, 2025 by copybara-service bot Loading…
@abhinavclemson
[TEST] Add gpu suppoprt in maxtext package
#2442 opened Oct 2, 2025 by kanglant • Draft
4 tasks done
Migrate Gpt-OSS to NNX
#2441 opened Oct 2, 2025 by hsuan-lun-chiang Loading…
4 tasks done
3
Migrate Gemma3DecoderLayer and Gemma3ScannableBlock to NNX gemini-review pull ready
#2439 opened Oct 2, 2025 by RissyRan Loading…
4 tasks done
13
[DO NOT MERGE ] test - Cli change gemini-review
#2433 opened Oct 1, 2025 by RissyRan Loading…
4 tasks
1
ProTip! Adding no:label will show everything without a label.