Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
193 Open 1,473 Closed
193 Open 1,473 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

sharding validator Improvements
#1810 opened Jun 6, 2025 by wei879-100 Loading…
4 tasks
[WIP] Full Llama4 VisionModel
#1809 opened Jun 5, 2025 by hengtaoguo • Draft
4 tasks
Create goodput_utils.py pull ready
#1808 opened Jun 5, 2025 by SurbhiJainUSC Loading…
4 tasks done
@dipannita08
2
fix github clone link
#1807 opened Jun 5, 2025 by junjieqian Loading…
4 tasks done
add support for cudnn sdpa checkpoint pull ready
#1805 opened Jun 4, 2025 by Cjkkkk Loading…
4 tasks done
2
I've added unit tests for critical conversion, layer, and inference c…
#1804 opened Jun 4, 2025 by stingram • Draft
1 of 4 tasks
Fix: Handle UnboundLocalError after load_next_batch() exception
#1803 opened Jun 4, 2025 by SurbhiJainUSC Loading…
4 tasks done
1
[NVIDIA] Add packing support for GPU flash attention
#1802 opened Jun 4, 2025 by kocchop Loading…
4 tasks done
Support Multi-Source Blending in Grain Data Pipeline
#1801 opened Jun 3, 2025 by bzantium Loading…
4 tasks
1
[DRAFT] v7x AOT support
#1796 opened May 31, 2025 by shralex Loading…
4 tasks done
[WIP] Layer-wise unit test without PyTorch
#1793 opened May 29, 2025 by hengtaoguo • Draft
4 tasks
Test Buildkit runner
#1791 opened May 29, 2025 by quoctruong • Draft
4 tasks done
Migrating Data Processing tests to CPU runners
#1787 opened May 29, 2025 by Rohan-Bierneni • Draft
4 tasks done
@Rohan-Bierneni
Gemma2 MaxText to HuggingFace checkpoint conversion
#1785 opened May 28, 2025 by YixuanWang-99 Loading…
4 tasks done
@RissyRan @hengtaoguo @khatwanimohit
45
Migrating docker build Github Actions for unit tests and daily image builds to GKE runners
#1783 opened May 27, 2025 by Rohan-Bierneni • Draft
3 tasks done
@Rohan-Bierneni
Internal Change
#1775 opened May 23, 2025 by copybara-service bot Loading…
2
feat: Add benchmark for inference request performance (TTFT, TPOT, th…
#1773 opened May 23, 2025 by yuyanpeng-google Loading…
4 tasks done
Elastic Training Fast Resume
#1769 opened May 22, 2025 by lukebaumann • Draft
4 tasks
tighten up the buffer size for ra2a during expert sharding.
#1767 opened May 22, 2025 by gpolovets1 Loading…
4 tasks done
5
Add deekseek distill golden logits generation.
#1758 opened May 19, 2025 by wang2yn84 Loading…
4 tasks done
11
[WIP] preprocessing for llama4
#1756 opened May 19, 2025 by hengtaoguo • Draft
4 tasks
[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
10
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
ProTip! Mix and match filters to narrow down what you’re looking for.