Skip to content

Pull requests: apple/axlearn

New pull request New
48 Open 850 Closed
48 Open 850 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Legacy flash remat fix
#943 opened Jan 23, 2025 by hanzhi713 Loading…
Uses pytest markers instead of module skip.
#941 opened Jan 22, 2025 by markblee • Draft
Add GKE A3 Ultra support
#940 opened Jan 22, 2025 by samos123 • Draft
5
Flash Attention for Neuron
#939 opened Jan 21, 2025 by apoorvtintin Loading…
16
Adds mesh rule for a3-megagpu-8g.
#936 opened Jan 20, 2025 by markblee Loading…
Enable GCP Workload Monitoring
#932 opened Jan 17, 2025 by Perseus14 • Draft
Allow external positions to be inputed in RoPE embedding layer
#926 opened Jan 15, 2025 by Firenze11 Loading…
16
Enabled running Pallas Flash Attention on CPU.
#922 opened Jan 14, 2025 by ds-hwang Loading…
16
[DO-NOT-MERGE] PR encompassing all changes needed to support neuron on Axlearn
#919 opened Jan 13, 2025 by apoorvtintin Loading…
1
Draft: Refactor JobSet for Pathways
#918 opened Jan 10, 2025 by jiya-zhang • Draft
26
TRN2 Meshes and Configurations
#916 opened Jan 10, 2025 by apoorvtintin Loading…
28
Enable cudnn attention dropout
#913 opened Jan 8, 2025 by hanzhi713 Loading…
2
use "true" and "false" instead of 0 and 1
#890 opened Dec 12, 2024 by samos123 Loading…
12
Input batch sharding strategy BATCH
#884 opened Dec 11, 2024 by apoorvtintin Loading…
20
Docker: Upgrade Jax to 0.4.37
#880 opened Dec 10, 2024 by samos123 • Draft
Optimizer offloading through weight-only offload
#867 opened Dec 4, 2024 by hanzhi713 Loading…
47
Integrate Orbax's emergency checkpoint.
#820 opened Nov 7, 2024 by hanzhi713 • Draft
Add Goodput & Badput recording and monitoring support.
#783 opened Oct 25, 2024 by dipannita08 Loading…
5 tasks done
37
restore test_apply_paddings_check runtime_checks test
#771 opened Oct 22, 2024 by mattjj Loading…
5
add Fuji v3 405b and solve HBM OOMs for larger models
#766 opened Oct 21, 2024 by samos123 • Draft
17
Alternative method of submitting jobs to DF Runner
#756 opened Oct 17, 2024 by remylouisew Loading…
4
Converts checkpointer and state builder to configurable.
#751 opened Oct 14, 2024 by markblee • Draft
Create custom jobset for pathways
#746 opened Oct 9, 2024 by jesus-orozco • Draft
5
Use regex for parsing step_dir
#739 opened Oct 7, 2024 by nlusskin Loading…
4
Add --megascale_abort_on_hangs flag for multi-slice TPU jobs
#731 opened Oct 3, 2024 by mugithi Loading…
12
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.