Skip to content

Pull requests: apple/axlearn

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

make spectrogram augmenter optional
#257 by zhiyun was closed Jan 5, 2024 Loading…
Do not require CI approval on main branch
#247 by madrob was merged Dec 13, 2023 Loading…
Lazy import tfio.
#233 by markblee was merged Dec 7, 2023 Loading…
Generalizes EncoderDecoderModel API.
#230 by markblee was merged Dec 11, 2023 Loading…
2
Update config.yml [DO NOT MERGE]
#226 by madrob was closed Dec 5, 2023 Loading…
Approval for CI
#225 by madrob was merged Dec 5, 2023 Loading…
Increases FD limit to 100000.
#222 by ruomingp was merged Dec 13, 2023 Loading…
2
Adds debug logging to gcp_settings that can be enabled with --verbosity=2.
#219 by ruomingp was closed Dec 13, 2023 Loading…
2
Relax jax version in dependencies.
#205 by markblee was closed Nov 27, 2023 Loading…
1
Introduces SpmdTrainer._train_step_input_partition_specs()
#202 by ruomingp was merged Nov 25, 2023 Loading…
update dataflow launch script in docstring
#196 by zhiyun was closed Dec 5, 2023 Loading…
Adds a bastion preemption command.
#182 by markblee was merged Nov 16, 2023 Loading…
Explains that watchdog being triggered is not an error in itself
#179 by ruomingp was merged Nov 13, 2023 Loading…
Makes axlearn ... config list print the current active project settings.
#177 by ruomingp was merged Nov 13, 2023 Loading…
Adds the axlearn gcp bastion history command.
#176 by ruomingp was merged Nov 13, 2023 Loading…
Add ASR encoder layers.
#174 by markblee was merged Nov 13, 2023 Loading…
Cleanup output_dim handling in subsampler.
#173 by markblee was merged Nov 11, 2023 Loading…
Moves the utils.host_to_global_device_array call from SpmdTrainer._run_step to SpmdTrainer.run.
#167 by ruomingp was merged Nov 7, 2023 Loading…
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
adding watchdog
#159 by kartikperisetla was closed Nov 3, 2023 • Draft
Adds GroupedQueryAttention.
#154 by markblee was merged Oct 30, 2023 Loading…
1
Raises a ValueError if the TPU version is unknown.
#153 by ruomingp was merged Oct 29, 2023 Loading…
More robust handling of unknown states.
#150 by markblee was merged Oct 29, 2023 Loading…
Check for 'WAITING_FOR_RESOURCES' state.
#144 by markblee was merged Oct 25, 2023 Loading…
ProTip! Updated in the last three days: updated:>2025-01-15.