Skip to content

Pull requests: apple/axlearn

New pull request New
Clear current search query, filters, and sorts
4 Open 7 Closed
4 Open 7 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

Flash Attention for Neuron
#939 opened Jan 21, 2025 by apoorvtintin Loading…
23
[DO-NOT-MERGE] PR encompassing all changes needed to support neuron on Axlearn
#919 opened Jan 13, 2025 by apoorvtintin Loading…
1
TRN2 Meshes and Configurations
#916 opened Jan 10, 2025 by apoorvtintin Loading…
43
Special remat for Neuron
#898 by apoorvtintin was merged Jan 14, 2025 Loading…
78
[DO-NOT-MERGE] PR encompassing all changes needed to support neuron on Axlearn
#886 by apoorvtintin was closed Jan 13, 2025 Loading…
1
Add meshes and config for TRN2/1 for Fuji models
#885 by apoorvtintin was closed Jan 10, 2025 Loading…
9
Input batch sharding strategy BATCH
#884 opened Dec 11, 2024 by apoorvtintin Loading…
20
Flash Attention for Neuron
#883 by apoorvtintin was closed Jan 21, 2025 Loading…
9
New DataPartitionType DATA
#567 by apoorvtintin was closed Dec 11, 2024 Loading…
6
Neuron support in Axlearn
#566 by apoorvtintin was closed Jan 12, 2025 Loading…
16
Gradient Accumulation in Axlearn
#465 by apoorvtintin was closed Jul 31, 2024 Loading…
12
ProTip! Add no:assignee to see everything that’s not assigned.