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

expose max_decode_len and eos_token_id in decoding
#328 opened Feb 17, 2024 by gyin94 Loading… updated Jul 2, 2024
5
Ulimit try
#260 opened Dec 20, 2023 by wwu137 • Draft updated Jul 2, 2024
Wwang48 rm
#229 opened Dec 6, 2023 by iamweiwang • Draft updated Jul 2, 2024
Adding the interpolation function for PosEmbeddingConverter
#139 opened Oct 23, 2023 by zbwglory Loading… updated Jul 2, 2024
3
Improve : Optimize Dockerfile for Better Build Efficiency
#126 opened Oct 16, 2023 by YashSalunke12 Loading… updated Jul 2, 2024
5
Flush vertexai tensorboard more frequently to workaround OutOfSpace bug.
#393 opened Apr 4, 2024 by markblee • Draft updated Jul 2, 2024
Create CONTRIBUTING.md
#149 opened Oct 28, 2023 by 0Armaan025 Loading… updated Jul 7, 2024
3
[Feature]Fixed: Added contributors section to readme.md file.
#151 opened Oct 29, 2023 by Kalyanimhala Loading… updated Jul 7, 2024
2
fixed regular expressions to support Unicode and improve readability
#267 opened Dec 25, 2023 by eukub Loading… updated Jul 7, 2024
1
Print step time for each step
#361 opened Mar 9, 2024 by samos123 Loading… updated Jul 7, 2024
7
Add labels for vm tier
#540 opened Jun 25, 2024 by eshen1991 Loading… updated Jul 7, 2024
5
Add log_every_n_steps in training wip
#523 opened Jun 17, 2024 by gyin94 Loading… updated Jul 9, 2024
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 updated Aug 12, 2024
remove nodeSelector provisioner-nodepool
#620 opened Aug 2, 2024 by samos123 • Draft updated Aug 28, 2024
1
1
Fix submission of Dataflow jobs
#711 opened Sep 24, 2024 by damccorm Loading… updated Sep 30, 2024
2
ssm_enhancement
#689 opened Sep 6, 2024 by vishesh9131 • Draft updated Oct 6, 2024
12
Converts checkpointer and state builder to configurable.
#751 opened Oct 14, 2024 by markblee • Draft updated Oct 14, 2024
Use regex for parsing step_dir
#739 opened Oct 7, 2024 by nlusskin Loading… updated Oct 15, 2024
4
Adding support for Pathways proxy
#690 opened Sep 9, 2024 by jesus-orozco Loading… updated Nov 18, 2024
22
add Fuji v3 405b and solve HBM OOMs for larger models
#766 opened Oct 21, 2024 by samos123 • Draft updated Dec 2, 2024
17
Create custom jobset for pathways
#746 opened Oct 9, 2024 by jesus-orozco • Draft updated Dec 9, 2024
5
Flash Attention for Neuron
#883 opened Dec 11, 2024 by apoorvtintin Loading… updated Dec 14, 2024
7
Docker: Upgrade Jax to 0.4.37
#880 opened Dec 10, 2024 by samos123 • Draft updated Dec 18, 2024
Alternative method of submitting jobs to DF Runner
#756 opened Oct 17, 2024 by remylouisew Loading… updated Jan 6, 2025
4
Add Goodput & Badput recording and monitoring support.
#783 opened Oct 25, 2024 by dipannita08 Loading… updated Jan 6, 2025
5 tasks done
33
ProTip! no:milestone will show everything without a milestone.