Skip to content

Actions: apple/axlearn

Actions

All workflows

Actions

Loading...
Loading

Showing runs from all workflows
114 workflow runs
114 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Fix RLHF slowdown in attention multi steps extend_step.
build-and-test #23: Pull request #849 opened by ds-hwang
November 19, 2024 16:27 21m 26s ds-hwang:mult_bug_fix
November 19, 2024 16:27 21m 26s
Conv1D supports paddings. (#847)
build-and-test #22: Commit 420ed7a pushed by github-merge-queue bot
November 19, 2024 02:01 21m 9s main
November 19, 2024 02:01 21m 9s
Optimize TPU Flash Attention (400x speed-up on 32k long context)
build-and-test #20: Pull request #845 synchronize by ds-hwang
November 19, 2024 01:16 16m 44s ds-hwang:flsh_op
November 19, 2024 01:16 16m 44s
Optimize TPU Flash Attention (400x speed-up on 32k long context)
pre-commit #16: Pull request #845 synchronize by ds-hwang
November 19, 2024 01:16 11m 15s ds-hwang:flsh_op
November 19, 2024 01:16 11m 15s
add Fuji v3 405b and solve HBM OOMs for larger models
build-and-test #19: Pull request #766 synchronize by samos123
November 19, 2024 00:58 Action required samos123:llama-3.1-405b
November 19, 2024 00:58 Action required
add Fuji v3 405b and solve HBM OOMs for larger models
pre-commit #15: Pull request #766 synchronize by samos123
November 19, 2024 00:58 Action required samos123:llama-3.1-405b
November 19, 2024 00:58 Action required
add Fuji v3 405b and solve HBM OOMs for larger models
build-and-test #18: Pull request #766 synchronize by samos123
November 19, 2024 00:40 Action required samos123:llama-3.1-405b
November 19, 2024 00:40 Action required
add Fuji v3 405b and solve HBM OOMs for larger models
pre-commit #14: Pull request #766 synchronize by samos123
November 19, 2024 00:40 Action required samos123:llama-3.1-405b
November 19, 2024 00:40 Action required
Integrate Orbax's emergency checkpoint.
pre-commit #13: Pull request #820 synchronize by hanzhi713
November 19, 2024 00:37 30m 31s hanzhi713:in-mem-ckpt
November 19, 2024 00:37 30m 31s
Integrate Orbax's emergency checkpoint.
build-and-test #17: Pull request #820 synchronize by hanzhi713
November 19, 2024 00:37 22m 11s hanzhi713:in-mem-ckpt
November 19, 2024 00:37 22m 11s
add Fuji v3 405b and solve HBM OOMs for larger models
build-and-test #16: Pull request #766 synchronize by samos123
November 19, 2024 00:20 Action required samos123:llama-3.1-405b
November 19, 2024 00:20 Action required
add Fuji v3 405b and solve HBM OOMs for larger models
pre-commit #12: Pull request #766 synchronize by samos123
November 19, 2024 00:20 Action required samos123:llama-3.1-405b
November 19, 2024 00:20 Action required
add Fuji v3 405b and solve HBM OOMs for larger models
build-and-test #15: Pull request #766 synchronize by samos123
November 19, 2024 00:03 Action required samos123:llama-3.1-405b
November 19, 2024 00:03 Action required
add Fuji v3 405b and solve HBM OOMs for larger models
pre-commit #11: Pull request #766 synchronize by samos123
November 19, 2024 00:03 Action required samos123:llama-3.1-405b
November 19, 2024 00:03 Action required
add Fuji v3 405b and solve HBM OOMs for larger models
build-and-test #14: Pull request #766 synchronize by samos123
November 19, 2024 00:00 Action required samos123:llama-3.1-405b
November 19, 2024 00:00 Action required
add Fuji v3 405b and solve HBM OOMs for larger models
pre-commit #10: Pull request #766 synchronize by samos123
November 19, 2024 00:00 Action required samos123:llama-3.1-405b
November 19, 2024 00:00 Action required
Optimize TPU Flash Attention (400x speed-up on 32k long context)
pre-commit #9: Pull request #845 synchronize by ds-hwang
November 18, 2024 23:52 11m 13s ds-hwang:flsh_op
November 18, 2024 23:52 11m 13s
Optimize TPU Flash Attention (400x speed-up on 32k long context)
build-and-test #13: Pull request #845 synchronize by ds-hwang
November 18, 2024 23:52 16m 43s ds-hwang:flsh_op
November 18, 2024 23:52 16m 43s
add Fuji v3 405b and solve HBM OOMs for larger models
build-and-test #12: Pull request #766 synchronize by samos123
November 18, 2024 23:43 Action required samos123:llama-3.1-405b
November 18, 2024 23:43 Action required
add Fuji v3 405b and solve HBM OOMs for larger models
pre-commit #8: Pull request #766 synchronize by samos123
November 18, 2024 23:43 Action required samos123:llama-3.1-405b
November 18, 2024 23:43 Action required
Ensure iterators are saved in per-process dir. (#846)
build-and-test #11: Commit 2335d13 pushed by github-merge-queue bot
November 18, 2024 22:16 21m 43s main
November 18, 2024 22:16 21m 43s
Generalizes pre- and post-spmd setup with init modules.
build-and-test #9: Pull request #848 opened by markblee
November 18, 2024 22:00 25m 13s markblee:markblee/init_modules
November 18, 2024 22:00 25m 13s