Skip to content

Pull requests: apple/axlearn

New pull request New
29 Open 646 Closed
29 Open 646 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 CuDNN fused MHA kernel to axlearn
#705 opened Sep 18, 2024 by kelvin-zou Loading…
8
No-op when garbage collecting non-existent checkpoint dir.
#703 opened Sep 17, 2024 by markblee Loading…
Bow metrics
#699 opened Sep 12, 2024 by headmyshoulder Loading…
@markblee @fnan
@ruomingp @gyin94
64
Adding support for Pathways proxy
#690 opened Sep 9, 2024 by jesus-orozco • Draft
ssm_enhancement
#689 opened Sep 6, 2024 by vishesh9131 Loading…
12
Speed up Axlearn CI
#688 opened Sep 6, 2024 by soundway Loading…
4
Makes BoundedAsyncCheckpointManager control max_concurrent_gb by the size of local shards instead of global shards.
#649 opened Aug 12, 2024 by ruomingp • Draft
Add @no_side_effects.
#647 opened Aug 9, 2024 by apghml Loading…
set hostNetwork to True for TPUGKEJob
#641 opened Aug 8, 2024 by samos123 Loading…
4
Preventing datasets instantiation from pretrained configs
#639 opened Aug 8, 2024 by fauconnier Loading…
2
allow using on-demand instead of spot only
#622 opened Aug 3, 2024 by samos123 Loading…
1
5
remove nodeSelector provisioner-nodepool
#620 opened Aug 2, 2024 by samos123 • Draft
1
1
Add details to TPU jobs launch log
#600 opened Jul 22, 2024 by nathalietran Loading…
New DataPartitionType DATA
#567 opened Jul 1, 2024 by apoorvtintin Loading…
6
Neuron support in Axlearn
#566 opened Jul 1, 2024 by apoorvtintin Loading…
14
Add github action support
#542 opened Jun 26, 2024 by samos123 Loading…
4
Add labels for vm tier
#540 opened Jun 25, 2024 by eshen1991 Loading…
5
Add log_every_n_steps in training wip
#523 opened Jun 17, 2024 by gyin94 Loading…
4
Flush vertexai tensorboard more frequently to workaround OutOfSpace bug.
#393 opened Apr 4, 2024 by markblee • Draft
Print step time for each step
#361 opened Mar 9, 2024 by samos123 Loading…
7
expose max_decode_len and eos_token_id in decoding
#328 opened Feb 17, 2024 by gyin94 Loading…
5
fixed regular expressions to support Unicode and improve readability
#267 opened Dec 25, 2023 by eukub Loading…
1
Ulimit try
#260 opened Dec 20, 2023 by wwu137 • Draft
Wwang48 rm
#229 opened Dec 6, 2023 by iamweiwang • Draft
[Feature]Fixed: Added contributors section to readme.md file.
#151 opened Oct 29, 2023 by Kalyanimhala Loading…
2
ProTip! Type g i on any issue or pull request to go back to the issue listing page.