Skip to content

Pull requests: apple/axlearn

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

Fix v6e boolean flags
#889 by Ethanlm was merged Dec 13, 2024 Loading…
1
Add default compiler options for v6e
#887 by samos123 was merged Dec 12, 2024 Loading…
10
Support v6e (Trillium)
#879 by Ethanlm was merged Dec 11, 2024 Loading…
[bug fix] fix issues with tool use metrics
#878 by fnan was merged Dec 10, 2024 Loading…
Makes Bastion write scheduler history files
#877 by ruomingp was merged Dec 10, 2024 Loading…
DiT: implement init_states and extend_step for DiT transformer
#876 by ds-hwang was merged Dec 10, 2024 Loading…
2
pin scikit-learn
#875 by fnan was merged Dec 9, 2024 Loading…
Refactor attention bias/mask/segments.
#873 by apghml was merged Dec 9, 2024 Loading…
2
Simplifies ScheduleResults.job_verdicts
#872 by ruomingp was merged Dec 9, 2024 Loading…
8
add tool use intent metrics
#871 by fnan was merged Dec 6, 2024 Loading…
7
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…
Support orbax state builder
#866 by hanzhi713 was merged Dec 6, 2024 Loading…
1
Refactoring Convolution; factor out convolutions from layers.py
#865 by ds-hwang was merged Dec 2, 2024 Loading…
1
Fix feedforward remat point
#864 by hanzhi713 was merged Dec 2, 2024 Loading…
Addresses Github Action warning.
#862 by markblee was merged Nov 30, 2024 Loading…
Generalizes decoder by taking input batch.
#861 by markblee was merged Nov 30, 2024 Loading…
Unify init and prefill for attention layers.
#860 by markblee was merged Nov 27, 2024 Loading…
Mainline upstream boilerplate intermediate
#859 by aws-mengchiy was closed Nov 27, 2024 • Draft
3
[Bug Fix] Fix a nit issue for Jax rollback
#856 by kelvin-zou was merged Nov 25, 2024 Loading…
[Bug fix] Jax rollback for TPU
#855 by kelvin-zou was closed Nov 23, 2024 Loading…
Rollback to Jax 0.4.33
#854 by kelvin-zou was merged Nov 22, 2024 Loading…
1
Implement ConvXDTranspose
#853 by ds-hwang was merged Nov 25, 2024 Loading…
7
Fix health check after https://github.com/apple/axlearn/pull/848
#852 by hanzhi713 was merged Dec 2, 2024 Loading…
Fix RLHF slowdown in attention multi steps extend_step.
#849 by ds-hwang was merged Nov 19, 2024 Loading…
2
ProTip! Exclude everything labeled bug with -label:bug.