Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
189 Open 1,422 Closed
189 Open 1,422 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[MaxText/inference_microbenchmark.py] Ensure correct exit code
#1753 opened May 18, 2025 by SamuelMarks Loading…
4 tasks done
1
Add recipe for knowledge distillation
#1752 opened May 16, 2025 by SurbhiJainUSC Loading…
4 tasks done
3
Internal change
#1751 opened May 16, 2025 by copybara-service bot Loading…
@gobbleturk
[*.py] Resolve pylint w-class: W0102,W0107,W0212,W0221,W0223,W0237,W0404,W0611,W0612,W0621,W0622,W0631,W0707,W0718,W1201,W1203,W1309,W1514,W4901 ; [code_style.sh,.github/workflows/CPUTests.yml] Enable w-class
#1749 opened May 16, 2025 by SamuelMarks Loading…
4 tasks done
18
Add max-prefill-length argument in distillation dataset generation script pull ready
#1748 opened May 15, 2025 by SurbhiJainUSC Loading…
4 tasks done
14
Add tokenizer's chat template to SFT
#1746 opened May 15, 2025 by SurbhiJainUSC Loading…
4 tasks done
9
Convert DenseGeneral to NNX
#1743 opened May 14, 2025 by bvandermoon Loading…
4 tasks done
Add deepseek c4 convergence test and v5p recipe pull ready
#1741 opened May 14, 2025 by suexu1025 Loading…
4 tasks done
13
add multimodal projector for llama4
#1733 opened May 13, 2025 by ZhaoyueCheng Loading…
4 tasks done
@ZhaoyueCheng @aireenmei
@hengtaoguo @yixinshi @jrplatin @vipannalla @gagika @mitalisi
2
[Don't open me] Internal testing v2
#1732 opened May 13, 2025 by copybara-service bot Loading…
@gobbleturk
Adding ml_goodput_measurement package to jax stable stack requirements
#1731 opened May 13, 2025 by Perseus14 Loading…
4 tasks done
9
convert RMSNorm to NNX
#1728 opened May 13, 2025 by cgarciae • Draft
4 tasks done
1
Convert MlpBlock to NNX
#1726 opened May 12, 2025 by cgarciae • Draft
Support vision attention for Llama4
#1724 opened May 12, 2025 by aireenmei Loading…
4 tasks done
@ZhaoyueCheng @hengtaoguo @gagika
9
Add comment that docker_upload_runner script needs bash version >= 4.2
#1717 opened May 11, 2025 by wstcliyu Loading…
4 tasks done
2
Inference Support in sparse_matmul
#1712 opened May 9, 2025 by gpolovets1 Loading…
4 tasks done
34
[Draft] Add Support for Headless Mode Recipe
#1708 opened May 9, 2025 by SujeethJinesh • Draft
4 tasks
Fix bad synthetic dataloader with per device batch size < 1.
#1706 opened May 9, 2025 by wang2yn84 Loading…
4 tasks done
1
[Draft] Add dedicated code owners for codebase
#1690 opened May 6, 2025 by RissyRan Loading…
4 tasks done
6
Add support for SPMD PP for deepseek decoder block
#1687 opened May 5, 2025 by gobbleturk Loading…
4 tasks done
@RissyRan @bvandermoon
13
Anisha from pretrained
#1684 opened May 3, 2025 by A9isha • Draft
4 tasks
Prepare for methodological pylint resolutions
#1667 opened May 1, 2025 by SamuelMarks Loading…
4 tasks done
@SamuelMarks
JetStream Offline Engine
#1664 opened Apr 30, 2025 by wenxindongwork Loading…
4 tasks done
@wenxindongwork
4
Internal change to fix copybara
#1662 opened Apr 30, 2025 by copybara-service bot Loading…
@gobbleturk
Internal change to fix import capitilization
#1661 opened Apr 30, 2025 by copybara-service bot Loading…
@gobbleturk
ProTip! Updated in the last three days: updated:>2025-05-15.