Skip to content

Pull requests: apple/axlearn

New pull request New
43 Open 713 Closed
43 Open 713 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

Weight only offload
#789 opened Oct 28, 2024 by hanzhi713 • Draft
Add multislice XLA flag to improve tf summary metric induced latency.
#787 opened Oct 28, 2024 by tgunter Loading…
Add TPU Monitoring for faster hang detection
#786 opened Oct 28, 2024 by kelvin-zou Loading…
6
improve groupnorm
#785 opened Oct 28, 2024 by berlino Loading…
Adds grain input dispatch.
#784 opened Oct 25, 2024 by markblee Loading…
Add Goodput & Badput recording and monitoring support.
#783 opened Oct 25, 2024 by dipannita08 Loading…
4 of 5 tasks
7
Support Causal Convolution.
#780 opened Oct 25, 2024 by ds-hwang • Queued
2
restore test_apply_paddings_check runtime_checks test
#771 opened Oct 22, 2024 by mattjj Loading…
5
add Fuji v3 405b model config
#766 opened Oct 21, 2024 by samos123 • Draft
8
Optimizer offloading draft and experiments
#757 opened Oct 18, 2024 by hanzhi713 • Draft
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
Use regex for parsing step_dir
#739 opened Oct 7, 2024 by nlusskin Loading…
4
Fix module import 2
#737 opened Oct 6, 2024 by vishesh9131 Loading…
12
Add --megascale_abort_on_hangs flag for multi-slice TPU jobs
#731 opened Oct 3, 2024 by mugithi Loading…
12
Upgrade jax to 0.4.33 for A3 Mega
#730 opened Oct 3, 2024 by samos123 • Draft
1
Set TF_FORCE_GPU_ALLOW_GROWTH=true by default
#712 opened Sep 24, 2024 by samos123 Loading…
2
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…
18
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
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
allow using on-demand instead of spot only
#622 opened Aug 3, 2024 by samos123 Loading…
1
5
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.