Skip to content

Pull requests: AI-Hypercomputer/maxtext

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

[WIP] llama4 ckpt conversion
#1816 opened Jun 10, 2025 by aireenmei • Draft
4 tasks
Refactor: Recording and logging training and evaluation metrics in all trainers
#1815 opened Jun 10, 2025 by SurbhiJainUSC Loading…
4 tasks done
@richjames0 @A9isha
Add total memory usage in GB and unit tests for activation offload
#1813 opened Jun 10, 2025 by zhenying-liu Loading…
5
Fixing autoregressive generation for local sliding window and chunk attention
#1811 opened Jun 9, 2025 by gagika Loading…
4 tasks done
@RissyRan @jrplatin
@gobbleturk @khatwanimohit
1
sharding validator Improvements
#1810 opened Jun 6, 2025 by wei879-100 Loading…
4 tasks done
@gobbleturk @wei879-100
11
[WIP] Full Llama4 VisionModel
#1809 opened Jun 5, 2025 by hengtaoguo • Draft
4 tasks
fix github clone link
#1807 opened Jun 5, 2025 by junjieqian Loading…
4 tasks done
4
add support for cudnn sdpa checkpoint pull ready
#1805 opened Jun 4, 2025 by Cjkkkk Loading…
4 tasks done
3
I've added unit tests for critical conversion, layer, and inference c…
#1804 opened Jun 4, 2025 by stingram • Draft
1 of 4 tasks
[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
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
Elastic Training Fast Resume
#1769 opened May 22, 2025 by lukebaumann • Draft
4 tasks done
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! Find all pull requests that aren't related to any open issues with -linked:issue.