Skip to content

Pull requests: apple/axlearn

New pull request New
59 Open 1,186 Closed
59 Open 1,186 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 page_pool argument
#1303 opened Jul 16, 2025 by hanzhi713 • Draft
Add RAttention
#1302 opened Jul 15, 2025 by berlino • Queued
[NOT FOR MERGE] Integrate all new Goodput features in v12
#1300 opened Jul 15, 2025 by dipannita08 • Draft
Sliding window KV cache supports the flash attention path via standard attention fallback.
#1299 opened Jul 15, 2025 by ds-hwang Loading…
2
[TEST, DO-NOT-MERGE] Trigger dependency problem between apache-beam and new tensorboard-plugin-profile.
#1295 opened Jul 14, 2025 by nataren Loading…
[DRAFT] Scale Test Configurations
#1293 opened Jul 11, 2025 by SujeethJinesh • Draft
Upgrade Tensorboard Plugin Profile to enable Roofline and Framework Ops Stats views.
#1292 opened Jul 11, 2025 by nataren Loading…
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…
2
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
Revert "Upgrade orbax checkpointer to 0.11.15"
#1281 opened Jul 1, 2025 by findmyway Loading…
[Do not review] Upgrade Python to 3.12 and use Ubuntu 24.04
#1280 opened Jul 1, 2025 by andersensam • Draft
Workload Hang & Rolling Window Goodput Monitoring Support
#1278 opened Jun 27, 2025 by dipannita08 Loading…
@findmyway
32
Update NCCL configs for A3 Ultra and A4
#1265 opened Jun 17, 2025 by andersensam Loading…
1
Update vm image.
#1262 opened Jun 14, 2025 by markblee Loading…
2
Adding basic elastic training (pause-and-resume)
#1256 opened Jun 12, 2025 by lukebaumann • Draft
12
CompositeLossMetrics now performs a weighted sum of losses.
#1251 opened Jun 10, 2025 by ds-hwang Loading…
3
Pathways: reuse setup_spmd for pathways init
#1248 opened Jun 9, 2025 by samos123 Loading…
6
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…
[UPDATE JAX API] Modify all the calls to jax.tree
#1206 opened May 22, 2025 by Steboss Loading…
ProTip! Follow long discussions with comments:>50.