Skip to content

Pull requests: apple/axlearn

New pull request New
Clear current search query, filters, and sorts
8 Open 11 Closed
8 Open 11 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 remat checkpoints to host instead of TPU memory
#643 by samos123 was merged Aug 14, 2024 Loading…
19
Print step time for each step
#361 opened Mar 9, 2024 by samos123 Loading…
7
increase timeout from 300s to 900s
#362 by samos123 was closed Apr 8, 2024 Loading…
1
GKE GPU A3 with TCPX support
#517 by samos123 was merged Jun 29, 2024 Loading…
64
Add github action support
#542 by samos123 was closed Jan 9, 2025 Loading…
4
remove nodeSelector provisioner-nodepool
#620 opened Aug 2, 2024 by samos123 • Draft
1
1
allow using on-demand instead of spot only
#622 opened Aug 3, 2024 by samos123 Loading…
1
5
Enable support for Kueue for GKETPUJob
#623 by samos123 was merged Aug 13, 2024 Loading…
1
Set remat_spec.policy None for fuji v2 70B
#638 by samos123 was closed Aug 8, 2024 Loading…
7
WIP add fuji 65B and GCP GPU support
#357 by samos123 was closed Aug 2, 2024 • Draft
set hostNetwork to True for TPUGKEJob
#641 opened Aug 8, 2024 by samos123 Loading…
5
Set TF_FORCE_GPU_ALLOW_GROWTH=true by default
#712 opened Sep 24, 2024 by samos123 Loading…
2
Upgrade jax to 0.4.33 for A3 Mega
#730 by samos123 was closed Oct 29, 2024 • Draft
2
add Fuji v3 405b and solve HBM OOMs for larger models
#766 opened Oct 21, 2024 by samos123 • Draft
17
improve GCS perf: Change resource limit to request
#851 by samos123 was merged Jan 17, 2025 Loading…
4
Docker: Upgrade Jax to 0.4.37
#880 opened Dec 10, 2024 by samos123 • Draft
Add default compiler options for v6e
#887 by samos123 was merged Dec 12, 2024 Loading…
12
use "true" and "false" instead of 0 and 1
#890 opened Dec 12, 2024 by samos123 Loading…
12
fix broken apt install google-perftools
#917 by samos123 was merged Jan 14, 2025 Loading…
1
15
ProTip! Filter pull requests by the default branch with base:main.