Skip to content

Pull requests: apple/axlearn

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
Assignee
Filter by who’s assigned
Sort

Pull requests list

Convert timestamp to datetime.
#1148 opened May 1, 2025 by markblee Loading…
Save O_PROJ on Fuji 70B-v2 for TRN2
#1147 opened May 1, 2025 by apoorvtintin Loading…
Page Attention GPU Kernel Integration
#1137 opened Apr 28, 2025 by christ-tt Loading…
[Jax API update] Remove jax_spmd_mode
#1136 opened Apr 28, 2025 by Steboss Loading…
Add ROCm support
#1112 opened Apr 16, 2025 by akasharidas Loading…
Adds logging to aux loss collection.
#950 opened Jan 26, 2025 by ruomingp Loading…
use "true" and "false" instead of 0 and 1
#890 opened Dec 12, 2024 by samos123 Loading…
Use regex for parsing step_dir
#739 opened Oct 7, 2024 by nlusskin Loading…
Set TF_FORCE_GPU_ALLOW_GROWTH=true by default
#712 opened Sep 24, 2024 by samos123 Loading…
Fix submission of Dataflow jobs
#711 opened Sep 24, 2024 by damccorm Loading…
ssm_enhancement
#689 opened Sep 6, 2024 by vishesh9131 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…
ProTip! Filter pull requests by the default branch with base:main.