Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
178 Open 2,564 Closed
178 Open 2,564 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 option to skip first token during logits comparison
#2953 opened Jan 15, 2026 by gagika Loading…
4 tasks done
1
Add Qwen3 Audio Encoder
#2952 opened Jan 15, 2026 by aireenmei • Draft
4 tasks done
1
Move integration tests to tests/integration and unit tests to tests/unit
#2949 opened Jan 14, 2026 by SurbhiJainUSC • Draft
4 tasks done
1
Move test_assets and logits generation script inside tests directory gemini-review
#2948 opened Jan 14, 2026 by SurbhiJainUSC Loading…
4 tasks done
9
Support NNX checkpoint conversion to HF
#2946 opened Jan 14, 2026 by hengtaoguo Loading…
4 tasks done
1
Set local_devices from CUDA_VISIBLE_DEVICES in initialize_jax_for_gpu
#2945 opened Jan 14, 2026 by gabeweisz Loading…
4 tasks done
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
1
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
4
Feat/migrate maxengine to nnx
#2937 opened Jan 13, 2026 by ecnal-cienet • Draft
4 tasks done
[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
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
13
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! Mix and match filters to narrow down what you’re looking for.