Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
182 Open 2,552 Closed
182 Open 2,552 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

Set local_devices from CUDA_VISIBLE_DEVICES in initialize_jax_for_gpu
#2945 opened Jan 14, 2026 by gabeweisz Loading…
4 tasks done
1
Fix the default C4 dataset version to 3.1.0.
#2944 opened Jan 14, 2026 by copybara-service bot Loading…
1
replace uses of pltpu.repeat with jnp.tile in pallas.
#2943 opened Jan 14, 2026 by copybara-service bot Loading…
1
Enabling gradient clipping in sft.sft_trainer
#2942 opened Jan 14, 2026 by ChingTsai Loading…
4 tasks done
Fix RL config init
#2941 opened Jan 14, 2026 by xuefgu • Draft
4 tasks done
1
Fix fsdp2d ds logical rule
#2940 opened Jan 14, 2026 by NuojCheng Loading…
4 tasks done
2
replace uses of pltpu.repeat with jnp.tile in pallas.
#2939 opened Jan 13, 2026 by copybara-service bot Loading…
1
Fix rngs used for training
#2938 opened Jan 13, 2026 by xibinliu Loading…
4 tasks done
3
Feat/migrate maxengine to nnx
#2937 opened Jan 13, 2026 by ecnal-cienet • Draft
4 tasks done
Fix Codecov Coverage Issue
#2934 opened Jan 13, 2026 by Shuang-cnt Loading…
4 tasks done
4
[DO NO MERGE] Draft for sparse
#2933 opened Jan 13, 2026 by RissyRan • Draft
4 tasks
18
[DRAFT] Insert gathers draft Draft PR
#2932 opened Jan 13, 2026 by NuojCheng • Draft
4 tasks
1
Refactor model loading in distillation to remove Tunix adapter.
#2927 opened Jan 10, 2026 by gagika Loading…
4 tasks done
@richjames0 @sheng-li @Obliviour
1
Fix: Update Fine-tuning Tutorial and Structure
#2925 opened Jan 9, 2026 by Shuang-cnt Loading…
4 tasks done
21
Trigger unit tests for docker images upload
#2924 opened Jan 9, 2026 by xibinliu Loading…
4 tasks done
1
Add jupyter notebook tests to Github CI tests
#2921 opened Jan 8, 2026 by SurbhiJainUSC Loading…
4 tasks done
13
[Diloco] Diloco trainer
#2920 opened Jan 8, 2026 by khatwanimohit • Draft
4 tasks done
12
Insert all gather per repeats for pipeline parallelism draft Draft PR
#2919 opened Jan 7, 2026 by NuojCheng • Draft
4 tasks
1
Happy New Year | Copyright 2023–2026 Google LLC
#2916 opened Jan 7, 2026 by SamuelMarks Loading…
4 tasks done
1
[src/MaxText/pyconfig.py] New initialize_pydantic_with_maxtext_config function ; [src/MaxText/examples/rl_llama3_demo.ipynb] Use new function
#2915 opened Jan 7, 2026 by SamuelMarks Loading…
4 tasks done
1
[Diloco] add diloco related utils
#2913 opened Jan 6, 2026 by khatwanimohit Loading…
4 tasks done
1
Add Packing Support for Context Parallelism (Ring Attention)
#2906 opened Dec 31, 2025 by kocchop Loading…
4 tasks done
14
Fix the AddLabel workflow permission issue
#2899 opened Dec 29, 2025 by xibinliu Loading…
4 tasks done
2
Add fractional batch size search for MaxText estimator draft Draft PR gemini-review
#2891 opened Dec 27, 2025 by NuojCheng • Draft
4 tasks done
4
fix: update knowledge distill docs and implementation
#2887 opened Dec 24, 2025 by mesakhcienet Loading…
4 tasks done
5
ProTip! Updated in the last three days: updated:>2026-01-11.