Skip to content

Pull requests: apple/axlearn

New pull request New
42 Open 1,251 Closed
42 Open 1,251 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

Upgrade GitHub Actions for Node 24 compatibility
#1356 opened Dec 16, 2025 by salmanmkc Loading…
Upgrades version of Orbax/TF/TFIO to support HNS atomic folder rename
#1355 opened Nov 22, 2025 by deepikarajani24 Loading…
1
Add audio model adapters and improve SSM partition specs
#1352 opened Nov 3, 2025 by vishesh9131 Loading…
3
Lkcolocated test stale
#1350 opened Oct 7, 2025 by lkolluru05 • Draft
6
[DO NOT REVIEW] CI Test stale
#1346 opened Sep 27, 2025 by ruhanprasad • Draft
1
Improve pathways checkpoint load times stale
#1345 opened Sep 23, 2025 by samos123 • Draft
5
Add scheduler as a launch flag stale
#1342 opened Sep 16, 2025 by linxiulei Loading…
1
WIP: pathways checkpointing experiments stale
#1340 opened Sep 12, 2025 by samos123 • Draft
2
[JAX API] Updating TransferToMemoryKind and jax.experimental.pallas.triton
#1339 opened Sep 10, 2025 by Steboss Loading…
Fixed some unit tests that failed on jax0.6.2 python 3.12 stale
#1337 opened Aug 28, 2025 by eshen1991 Loading…
3
Updated the pathwaysutils version
#1335 opened Aug 26, 2025 by lukebaumann Loading…
Adds pause-resume elasticity if the proxy backend is used. stale
#1334 opened Aug 22, 2025 by lukebaumann Loading…
7
working colocated example stale
#1333 opened Aug 20, 2025 by lkolluru05 • Draft
1
Multi-tier checkpointing + orbax replicator stale
#1332 opened Aug 12, 2025 by ehorning Loading…
64
Muyang yu/lwx service copy stale
#1330 opened Aug 7, 2025 by muyangyuapple • Draft
2
Add Orbax Checkpoint Logger for Goodput stale
#1326 opened Aug 1, 2025 by dipannita08 • Draft
2
feat: add sigterm handling for graceful termination stale
#1324 opened Jul 29, 2025 by linxiulei • Draft
4
Generate requirement files and pyproject.toml for an AXLearn GPU workflow using seed-env stale
#1320 opened Jul 28, 2025 by kanglant • Draft
2
minimal example stale
#1318 opened Jul 25, 2025 by RsEnts • Draft
3
Update JAX API usage to latest version stale
#1317 opened Jul 25, 2025 by ReNothingg Loading…
9
[NOT FOR MERGE] Integrate all new Goodput features in v12 stale
#1300 opened Jul 15, 2025 by dipannita08 • Draft
2
[DRAFT] Scale Test Configurations stale
#1293 opened Jul 11, 2025 by SujeethJinesh • Draft
2
Add fuji 150b for testing stale
#1291 opened Jul 11, 2025 by hanzhi713 • Draft
2
gcsfuse_mount_spec parsing and supporting mount_options from command line stale
#1290 opened Jul 11, 2025 by deepikarajani24 Loading…
2
Pathways Updates stale
#1288 opened Jul 8, 2025 by lukebaumann Loading…
9
ProTip! Exclude everything labeled bug with -label:bug.