Skip to content

Pull requests: apple/axlearn

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

Remove version pin of typing-extensions
#904 by wangkuiyi was closed Jan 7, 2025 Loading…
3
Adds @apple/axlearn-admins to CODEOWNERS.
#896 by ruomingp was closed Dec 17, 2024 Loading…
1
[DO-NOT-MERGE] PR encompassing all changes needed to support neuron on Axlearn
#886 by apoorvtintin was closed Jan 13, 2025 Loading…
1
add merge_group to action workflows
#870 by madrob was merged Dec 6, 2024 Loading…
Enable GCP Workload Monitoring
#868 by Perseus14 was closed Dec 4, 2024 Loading…
Mainline upstream boilerplate intermediate
#859 by aws-mengchiy was closed Nov 27, 2024 • Draft
3
[Bug fix] Jax rollback for TPU
#855 by kelvin-zou was closed Nov 23, 2024 Loading…
Optimize TPU Flash Attention (400x speed-up on 32k long context)
#845 by ds-hwang was closed Jan 7, 2025 • Draft
A skeleton of shuffling tensorstore_specs for serialization/deserialization.
#833 by ruomingp was closed Nov 13, 2024 Loading…
1
[DO NOT CHECK IN, TEST ONLY]Test CI OOM issue
#792 by kelvin-zou was closed Oct 29, 2024 Loading…
Weight only offload
#789 by hanzhi713 was closed Nov 1, 2024 • Draft
2
tokens_per_batch fixed to take into account DP and micro-batch accumulation steps
#782 by amithrm was closed Oct 25, 2024 Loading…
Support more types of groupnorm
#772 by berlino was closed Oct 28, 2024 Loading…
3
Add neuron attention with tests
#769 by lipovsek-aws was closed Oct 22, 2024 Loading…
Optimizer offloading draft and experiments
#757 by hanzhi713 was closed Nov 26, 2024 • Draft
support ssd/mamba2
#747 by berlino was closed Oct 11, 2024 Loading…
Upgrade jax to 0.4.33 for A3 Mega
#730 by samos123 was closed Oct 29, 2024 • Draft
2
Adds initial lm training inputs with grain.
#727 by markblee was merged Oct 3, 2024 Loading…
2
Adds ASR WER calculator.
#726 by markblee was merged Oct 3, 2024 Loading…
1
Add --megascale_abort_on_hangs flag for multi-slice TPU jobs
#716 by mugithi was closed Oct 3, 2024 Loading…
13
add bert 768
#709 by YXSIO was closed Sep 23, 2024 Loading…
Fix weighted scalar division by zero.
#676 by markblee was merged Aug 27, 2024 Loading…
Style changes with py39+ as target.
#672 by miaojingang was closed Aug 23, 2024 • Draft
Simplify flatten_items.
#666 by markblee was merged Aug 27, 2024 Loading…
Adds support for private worker pools.
#665 by markblee was merged Aug 27, 2024 Loading…
ProTip! Follow long discussions with comments:>50.