Skip to content

Pull requests: apple/axlearn

New pull request New
38 Open 1,237 Closed
38 Open 1,237 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 fuji v3 405b config
#1338 opened Sep 4, 2025 by samos123 Loading…
Fixed some unit tests that failed on jax0.6.2 python 3.12
#1337 opened Aug 28, 2025 by eshen1991 Loading…
2
Updated the pathwaysutils version
#1335 opened Aug 26, 2025 by lukebaumann Loading…
Adds pause-resume elasticity if the proxy backend is used.
#1334 opened Aug 22, 2025 by lukebaumann Loading…
2
working colocated example
#1333 opened Aug 20, 2025 by lkolluru05 • Draft
Multi-tier checkpointing + orbax replicator
#1332 opened Aug 12, 2025 by ehorning • Draft
61
Muyang yu/lwx service copy
#1330 opened Aug 7, 2025 by muyangyuapple • Draft
Add Orbax Checkpoint Logger for Goodput
#1326 opened Aug 1, 2025 by dipannita08 • Draft
feat: add sigterm handling for graceful termination
#1324 opened Jul 29, 2025 by linxiulei • Draft
2
Don't review Orbax configs: Regular and Emergency
#1323 opened Jul 29, 2025 by samos123 • Draft
Generate requirement files and pyproject.toml for an AXLearn GPU workflow using seed-env
#1320 opened Jul 28, 2025 by kanglant • Draft
minimal example
#1318 opened Jul 25, 2025 by RsEnts • Draft
1
Update JAX API usage to latest version
#1317 opened Jul 25, 2025 by ReNothingg Loading…
7
[NOT FOR MERGE] Integrate all new Goodput features in v12
#1300 opened Jul 15, 2025 by dipannita08 • Draft
[DRAFT] Scale Test Configurations
#1293 opened Jul 11, 2025 by SujeethJinesh • Draft
Add fuji 150b for testing
#1291 opened Jul 11, 2025 by hanzhi713 • Draft
gcsfuse_mount_spec parsing and supporting mount_options from command line
#1290 opened Jul 11, 2025 by deepikarajani24 Loading…
Pathways Updates
#1288 opened Jul 8, 2025 by lukebaumann Loading…
8
Support empty folder deletion for HNS bucket with Orbax
#1286 opened Jul 7, 2025 by deepikarajani24 Loading…
23
changes added to test orbax
#1282 opened Jul 2, 2025 by lkolluru05 • Draft
Pathways: reuse setup_spmd for pathways init
#1248 opened Jun 9, 2025 by samos123 Loading…
7
orbax to reg checkpointer conversion
#1246 opened Jun 6, 2025 by lkolluru05 • Draft
4
Ensure GCS bucket region matches config region
#1220 opened May 28, 2025 by ehorning • Draft
1
Add DataflowRunnerJob to allow running Dataflow jobs via Bastion.
#1213 opened May 27, 2025 by zachberger • Draft
4 tasks
Updated jax.tree_map to jax.tree_util.tree_map in update_transformation.py
#1207 opened May 22, 2025 by m-ny Loading…
ProTip! Exclude everything labeled bug with -label:bug.