Skip to content

Pull requests: apple/axlearn

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

Expose segment_ids to _compute_attention in FlashAttention
#714 by changlan was merged Sep 28, 2024 Loading…
36 👍 3
Bump GCP and Dataflow deps.
#93 by markblee was merged Oct 3, 2023 Loading…
1 👍 1
Fix flaky test.
#631 by apghml was merged Aug 5, 2024 Loading…
👍 1
Fix feedforward remat point
#864 by hanzhi713 was merged Dec 2, 2024 Loading…
👍 1
Gradient Accumulation in Axlearn
#465 by apoorvtintin was closed Jul 31, 2024 Loading…
12 👍 1
Adds ASR WER calculator.
#726 by markblee was merged Oct 3, 2024 Loading…
1 👍 1
Add --megascale_abort_on_hangs flag for multi-slice TPU jobs
#716 by mugithi was closed Oct 3, 2024 Loading…
13 👍 1
Upgrade Jax to 0.4.25
#413 by kelvin-zou was merged Apr 23, 2024 Loading…
👍 1
Decouples from flax.
#374 by markblee was merged Mar 19, 2024 Loading…
👍 1
Support customized mesh rules to support different HWs
#696 by kelvin-zou was merged Sep 13, 2024 Loading…
39 👍 1
[rnnt] make blank_id field required in transducer
#419 by zhiyun was merged Apr 24, 2024 Loading…
👍 1
Updates checkpointer interface…
#635 by markblee was merged Aug 7, 2024 Loading…
👍 1
Adds convert_dense_to_moe_parameters .
#584 by ruomingp was merged Jul 21, 2024 Loading…
1 👍 1
Clean Unschedulable jobs.
#377 by apghml was merged Mar 23, 2024 Loading…
👍 1
No-op when garbage collecting non-existent checkpoint dir.
#703 by markblee was merged Sep 21, 2024 Loading…
👍 1
load llama v3 weights into fuji
#778 by sychen52 was merged Oct 28, 2024 Loading…
@sychen52
7 👍 1
Support SplashAttention.
#722 by apghml was merged Oct 2, 2024 Loading…
👍 1
Add support for tensor including UTF-8 string in inference_output
#23 by mialsy was merged Aug 3, 2023 Loading…
Add ReadOptions args to _make_autoregressive_inputs
#931 by RsEnts was merged Jan 17, 2025 Loading…
3
add EfficientNetV2 implementation
#14 by edaxberger was merged Aug 2, 2023 Loading…
1
ADD | Adding logging for checkpoint directory
#13 by fauconnier was merged Jul 27, 2023 Loading…
1
Remove hf_pretrained_loaders.py.
#12 by tuzhucheng was merged Jul 26, 2023 Loading…
@markblee
Minor cleanup and add batch_size to fake text source
#11 by SnehaNB was merged Jul 26, 2023 Loading…
Fix typo: os.kath -> os.path
#6 by llcao was merged Jul 20, 2023 Loading…
Fix some permlinks
#5 by xianzhidu was merged Jul 22, 2023 Loading…
1
ProTip! What’s not been updated in a month: updated:<2024-12-18.