Skip to content

Pull requests: apple/axlearn

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
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort

Pull requests list

Respect the memory kind of sharding in deserialization
#1231 opened May 31, 2025 by zlfben Loading…
Adds codeowners for axlearn/cloud.
#1230 opened May 30, 2025 by markblee Loading…
Update test flash attention test forward
#1226 opened May 29, 2025 by matthew-e-hopkins Loading…
Update pathways image to jax 0.5.3
#1225 opened May 29, 2025 by karan Loading…
Ensure GCS bucket region matches config region
#1220 opened May 28, 2025 by ehorning Loading…
[JAX API Update] Remove jax_spmd_mode from config
#1211 opened May 23, 2025 by Steboss Loading…
[UPDATE JAX API] Modify all the calls to jax.tree
#1206 opened May 22, 2025 by Steboss Loading…
[UPDATE JAX API] Update trainer_test _normalized_spec
#1201 opened May 21, 2025 by Steboss Loading…
Debug regression
#1196 opened May 21, 2025 by raytomato Draft
Jax 0.5.3
#1182 opened May 14, 2025 by matthew-e-hopkins Loading…
Adding LWS Integration
#1174 opened May 12, 2025 by Edwinhr716 Loading…
Save O_PROJ on Fuji 70B-v2 for TRN2
#1147 opened May 1, 2025 by apoorvtintin Loading…
Add ROCm support
#1112 opened Apr 16, 2025 by akasharidas Loading…
Adds logging to aux loss collection.
#950 opened Jan 26, 2025 by ruomingp Loading…
ProTip! Add no:assignee to see everything that’s not assigned.