Skip to content

Pull requests: apple/axlearn

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

Enable to assign different parameters dtype during training
#1037 opened Mar 5, 2025 by jialingt Loading…
Trainer: the model analysis on the AOT compiled JAX program.
#1036 opened Mar 5, 2025 by ds-hwang Loading…
7
[CI] Enable 4- and 8-device Flash Attention layers tests.
#1027 opened Feb 28, 2025 by liboyue Loading…
1
Improvements and fixes to gradient accumulation
#993 opened Feb 14, 2025 by apoorvtintin Loading…
28
Basic FP8 Training Support
#992 opened Feb 13, 2025 by hanzhi713 Loading…
16
Migrate from Legacy JAX APIs jax.tree_util to jax.tree
#986 opened Feb 12, 2025 by apivovarov Loading…
7
Use Accuracy from cross_entropy in causal_lm.py::CrossEntropyLossMetrics
#983 opened Feb 12, 2025 by apivovarov Loading…
3
Adds logging to aux loss collection.
#950 opened Jan 26, 2025 by ruomingp Loading…
1
Uses pytest markers instead of module skip.
#941 opened Jan 22, 2025 by markblee • Draft
Add GKE A3 Ultra support
#940 opened Jan 22, 2025 by samos123 Loading…
26
Enable GCP Workload Monitoring
#932 opened Jan 17, 2025 by Perseus14 Loading…
3
Draft: Refactor JobSet for Pathways
#918 opened Jan 10, 2025 by jiya-zhang • Draft
27
use "true" and "false" instead of 0 and 1
#890 opened Dec 12, 2024 by samos123 Loading…
13
restore test_apply_paddings_check runtime_checks test
#771 opened Oct 22, 2024 by mattjj Loading…
5
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…
4
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
Add --megascale_abort_on_hangs flag for multi-slice TPU jobs
#731 opened Oct 3, 2024 by mugithi Loading…
13
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…
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
ProTip! Type g i on any issue or pull request to go back to the issue listing page.