Skip to content

Pull requests: apple/axlearn

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

Add GKE A3 Mega, A3 Ultra, and A4 High instance support
#1097 by andersensam was closed Apr 10, 2025 Loading…
9
[Jax API update] Update jax.core.Primitive to jax.extend.core.Primitive
#1082 by Steboss was closed Apr 11, 2025 Loading…
13
Sync with upstream and add A4 support
#1060 by andersensam was closed Mar 18, 2025 • Draft
Reverts sliding window attention changes.
#1004 by markblee was merged Feb 19, 2025 Loading…
1
Fix Parameter Redefinition in learner_test.py
#988 by apivovarov was closed Feb 13, 2025 Loading…
6
Add GKE A3 Ultra support
#940 by samos123 was closed Apr 4, 2025 Loading…
44
Enable GCP Workload Monitoring
#932 by Perseus14 was closed Mar 18, 2025 Loading…
3
[DO-NOT-MERGE] PR encompassing all changes needed to support neuron on Axlearn
#919 by apoorvtintin was closed Mar 5, 2025 Loading…
2
Flash Attention for Neuron
#883 by apoorvtintin was closed Jan 21, 2025 Loading…
9
Docker: Upgrade Jax to 0.4.37
#880 by samos123 was closed Jan 25, 2025 • Draft
[Bug fix] Jax rollback for TPU
#855 by kelvin-zou was closed Nov 23, 2024 Loading…
remove nodeSelector provisioner-nodepool
#620 by samos123 was closed Apr 4, 2025 • Draft
1
2
Te flash attention
#464 by kelvin-zou was closed Sep 18, 2024 • Draft
Draft PR to add GoodPut Measurement
#370 by jiya-zhang was closed Aug 14, 2024 • Draft
Renames mask to live_targets for cross_entropy_loss and binary_cross_entropy_loss.
#329 by ruomingp was merged Feb 18, 2024 Loading…
2
ProTip! Filter pull requests by the default branch with base:main.