Skip to content

Pull requests: apple/axlearn

New pull request New
Clear current search query, filters, and sorts
26 Open 118 Closed
26 Open 118 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

Adds SpmdTrainer.Config.mesh_rules and --mesh_selector to override trainer mesh configuration.
#104 by ruomingp was merged Oct 8, 2023 Loading…
Replaces uses of utils.host_to_global_device_array with map_utils.host_local_array_to_global_array.
#115 by ruomingp was closed Oct 13, 2023 Loading…
upgrade transformers version
#121 by gyin94 was closed Oct 14, 2023 Loading…
1
Keep user-specified find_links in tpu_runner.py
#130 by ya5ut was closed Oct 18, 2023 • Draft
[DO-NOT-MERGE] PR encompassing all changes needed to support neuron on Axlearn
#886 by apoorvtintin was closed Jan 13, 2025 Loading…
1
Update README.md
#142 by FarukhS52 was closed Oct 25, 2023 Loading…
Check for 'WAITING_FOR_RESOURCES' state.
#144 by markblee was merged Oct 25, 2023 Loading…
adding watchdog
#159 by kartikperisetla was closed Nov 3, 2023 • Draft
Replace jax.random.KeyArray with jax.Array to suppress deprecation warnings.
#166 by markblee was merged Nov 4, 2023 Loading…
Moves conformer layers to axlearn/audio.
#165 by markblee was closed Nov 5, 2023 Loading…
1
Gradient Accumulation in Axlearn
#465 by apoorvtintin was closed Jul 31, 2024 Loading…
12
Change the CloudBuild to Sync by Default
#616 by amcw7777 was closed Jul 31, 2024 Loading…
WIP add fuji 65B and GCP GPU support
#357 by samos123 was closed Aug 2, 2024 • Draft
Cleanup output_dim handling in subsampler.
#173 by markblee was merged Nov 11, 2023 Loading…
Makes axlearn ... config list print the current active project settings.
#177 by ruomingp was merged Nov 13, 2023 Loading…
Add ASR encoder layers.
#174 by markblee was merged Nov 13, 2023 Loading…
Adds the axlearn gcp bastion history command.
#176 by ruomingp was merged Nov 13, 2023 Loading…
Adds a bastion preemption command.
#182 by markblee was merged Nov 16, 2023 Loading…
Update trainer config test_utils.
#17 by markblee was merged Jul 29, 2023 Loading…
Evaler policies.
#15 by markblee was merged Jul 28, 2023 Loading…
torch: attention softmax as float32 for bf16
#18 by gyin94 was closed Aug 3, 2023 Loading…
Adds the CLI.
#7 by markblee was merged Jul 21, 2023 Loading…
Update README.md
#9 by TanayShukla was closed Aug 5, 2023 Loading…
Fixes lion optimizer dtypes.
#36 by markblee was merged Aug 17, 2023 Loading…
Removes _auto_child_context_method_names.
#39 by markblee was merged Aug 22, 2023 Loading…
ProTip! Mix and match filters to narrow down what you’re looking for.