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

Lazy import tfio.
#233 by markblee was merged Dec 7, 2023 Loading…
Add Request ID to HTTP requests
#35 by jiya-zhang was closed Dec 14, 2023 Loading…
4
update dataflow script
#258 by zhiyun was closed Dec 19, 2023 Loading…
Generalizes EncoderDecoderModel API.
#230 by markblee was merged Dec 11, 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
Increases FD limit to 100000.
#222 by ruomingp was merged Dec 13, 2023 Loading…
2
Do not require CI approval on main branch
#247 by madrob was merged Dec 13, 2023 Loading…
[DO NOT MERGE] test CI
#262 by zhiyun was closed Dec 21, 2023 Loading…
make spectrogram augmenter optional
#257 by zhiyun was closed Jan 5, 2024 Loading…
[Will not merge] testing CI test triggering
#261 by zhzhyi was closed Jan 6, 2024 Loading…
Implements tfio writer for simple log rotation.
#289 by markblee was merged Jan 24, 2024 Loading…
Provide action to launchers, e.g. to skip bundle instantiation.
#307 by markblee was merged Feb 6, 2024 Loading…
add option to disable bundle local axlearn
#299 by gyin94 was closed Feb 6, 2024 Loading…
1
Adds --root_module to run_aot_compilation.py.
#292 by ruomingp was closed Feb 8, 2024 Loading…
Increases file descriptor limit for root to 100K in start_tpu.sh.
#259 by ruomingp was closed Feb 8, 2024 Loading…
4
Adds select_mesh_config call to run_aot_compilation.py.
#318 by ruomingp was merged Feb 13, 2024 Loading…
Fixes named_chain where some kwargs values are configs.
#337 by ruomingp was merged Feb 21, 2024 Loading…
Removes StartBastionJob, which seems unnecessary.
#325 by ruomingp was merged Feb 16, 2024 Loading…
Fixes named_chain.
#336 by ruomingp was merged Feb 21, 2024 Loading…
Adds initial lm training inputs with grain.
#727 by markblee was merged Oct 3, 2024 Loading…
2
Add --megascale_abort_on_hangs flag for multi-slice TPU jobs
#716 by mugithi was closed Oct 3, 2024 Loading…
13
Add Orbax Checkpointing
#627 by jiya-zhang was closed Aug 21, 2024 • Draft
13
Adds ASR WER calculator.
#726 by markblee was merged Oct 3, 2024 Loading…
1
increase timeout from 300s to 900s
#362 by samos123 was closed Apr 8, 2024 Loading…
1
Makes trainer use separate jit programs for summaries vs. not.
#440 by ruomingp was closed May 7, 2024 • Draft
ProTip! Updated in the last three days: updated:>2025-01-16.