Skip to content

Pull requests: apple/axlearn

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

[DO-NOT-MERGE] PR encompassing all changes needed to support neuron on Axlearn
#886 opened Dec 11, 2024 by apoorvtintin Loading…
Flash Attention for Neuron
#883 opened Dec 11, 2024 by apoorvtintin Loading…
7
Update RoformerQKVLinear to support kv_state
#882 opened Dec 11, 2024 by qdavid1 Loading…
2
Docker: Upgrade Jax to 0.4.37
#880 opened Dec 10, 2024 by samos123 • Draft
improve GCS perf: Change resource limit to request
#851 opened Nov 19, 2024 by samos123 Loading…
1
Add llama 3 tokenizer
#850 opened Nov 19, 2024 by sychen52 Loading…
25
Optimize TPU Flash Attention (400x speed-up on 32k long context)
#845 opened Nov 18, 2024 by ds-hwang • Draft
Add Mamab2 and its Jamba variant
#839 opened Nov 14, 2024 by berlino Loading…
2
Integrate Orbax's emergency checkpoint.
#820 opened Nov 7, 2024 by hanzhi713 • Draft
Add Goodput & Badput recording and monitoring support.
#783 opened Oct 25, 2024 by dipannita08 Loading…
5 tasks done
31
add Fuji v3 405b and solve HBM OOMs for larger models
#766 opened Oct 21, 2024 by samos123 • Draft
17
Alternative method of submitting jobs to DF Runner
#756 opened Oct 17, 2024 by remylouisew Loading…
3
Converts checkpointer and state builder to configurable.
#751 opened Oct 14, 2024 by markblee • Draft
Create custom jobset for pathways
#746 opened Oct 9, 2024 by jesus-orozco • Draft
5
Use regex for parsing step_dir
#739 opened Oct 7, 2024 by nlusskin Loading…
4
Fix submission of Dataflow jobs
#711 opened Sep 24, 2024 by damccorm Loading…
2
Adding support for Pathways proxy
#690 opened Sep 9, 2024 by jesus-orozco Loading…
22
ssm_enhancement
#689 opened Sep 6, 2024 by vishesh9131 • Draft
12
Makes BoundedAsyncCheckpointManager control max_concurrent_gb by the size of local shards instead of global shards.
#649 opened Aug 12, 2024 by ruomingp • Draft
remove nodeSelector provisioner-nodepool
#620 opened Aug 2, 2024 by samos123 • Draft
1
1
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
ProTip! What’s not been updated in a month: updated:<2024-11-13.