Skip to content

Actions: fattorib/ZeRO-transformer

Actions

All workflows

Actions

Loading...

Showing runs from all workflows
214 workflow runs
214 workflow runs
Event

Filter by event

Status

Filter by status

Branch
Actor

Filter by actor

pin jax version
Tests #1148: Commit c1e7ece pushed by fattorib
June 11, 2023 20:49 3m 53s main
June 11, 2023 20:49 3m 53s
einops cleanups
Tests #1147: Commit 9c14da6 pushed by fattorib
June 9, 2023 18:56 4m 24s main
June 9, 2023 18:56 4m 24s
clean git tree before rebase
Tests #1145: Commit a59612f pushed by fattorib
June 6, 2023 17:12 4m 31s pjit-tensor-parallel
June 6, 2023 17:12 4m 31s
use real step for resuming
Tests #1144: Commit b82b440 pushed by fattorib
June 6, 2023 17:09 4m 36s pjit-tensor-parallel
June 6, 2023 17:09 4m 36s
wrap in tuple
Tests #1143: Commit 423e946 pushed by fattorib
June 6, 2023 17:04 4m 6s pjit-tensor-parallel
June 6, 2023 17:04 4m 6s
spec change
Tests #1142: Commit ec335f7 pushed by fattorib
June 6, 2023 17:01 3m 32s pjit-tensor-parallel
June 6, 2023 17:01 3m 32s
donate argnums
Tests #1141: Commit 8d41bf5 pushed by fattorib
June 6, 2023 16:58 4m 7s pjit-tensor-parallel
June 6, 2023 16:58 4m 7s
with mesh
Tests #1140: Commit 33b40cb pushed by fattorib
June 6, 2023 16:53 4m 42s pjit-tensor-parallel
June 6, 2023 16:53 4m 42s
wrap all resumes in cpu context
Tests #1139: Commit 48ef1e3 pushed by fattorib
June 6, 2023 16:52 4m 6s pjit-tensor-parallel
June 6, 2023 16:52 4m 6s
pre-shard with pjit
Tests #1138: Commit fc7c8b7 pushed by fattorib
June 6, 2023 16:50 3m 37s pjit-tensor-parallel
June 6, 2023 16:50 3m 37s
move into context manager
Tests #1137: Commit 496adf4 pushed by fattorib
June 6, 2023 16:47 4m 10s pjit-tensor-parallel
June 6, 2023 16:47 4m 10s
use CPU context manager
Tests #1136: Commit 18e7f33 pushed by fattorib
June 6, 2023 16:45 4m 30s pjit-tensor-parallel
June 6, 2023 16:45 4m 30s
map params to CPU
Tests #1135: Commit 6a94620 pushed by fattorib
June 6, 2023 16:40 3m 34s pjit-tensor-parallel
June 6, 2023 16:40 3m 34s
arg name
Tests #1134: Commit dee496b pushed by fattorib
June 6, 2023 16:36 3m 43s pjit-tensor-parallel
June 6, 2023 16:36 3m 43s
manually restore checkpoint partitions
Tests #1133: Commit 5825f86 pushed by fattorib
June 6, 2023 16:25 3m 45s pjit-tensor-parallel
June 6, 2023 16:25 3m 45s
NamedSharding doesn't work in multihost
Tests #1132: Commit 69b38a9 pushed by fattorib
June 6, 2023 14:16 4m 12s pjit-tensor-parallel
June 6, 2023 14:16 4m 12s
create NamedShardings
Tests #1131: Commit b6e35d0 pushed by fattorib
June 6, 2023 14:08 3m 42s pjit-tensor-parallel
June 6, 2023 14:08 3m 42s
use jax.device_put
Tests #1130: Commit fde1cc4 pushed by fattorib
June 6, 2023 14:02 3m 47s pjit-tensor-parallel
June 6, 2023 14:02 3m 47s
manual deletions
Tests #1129: Commit 1c072c1 pushed by fattorib
June 6, 2023 13:56 4m 40s pjit-tensor-parallel
June 6, 2023 13:56 4m 40s
disable bucket clearing
Tests #1128: Commit 7b8986d pushed by fattorib
June 6, 2023 13:52 4m 36s pjit-tensor-parallel
June 6, 2023 13:52 4m 36s
override resume step
Tests #1127: Commit b8f3ecd pushed by fattorib
June 6, 2023 13:48 4m 43s pjit-tensor-parallel
June 6, 2023 13:48 4m 43s
donate argnums
Tests #1126: Commit 76cfc1c pushed by fattorib
June 6, 2023 13:43 4m 21s pjit-tensor-parallel
June 6, 2023 13:43 4m 21s
call params/opt to CPU first
Tests #1125: Commit d1e780f pushed by fattorib
June 6, 2023 13:40 3m 40s pjit-tensor-parallel
June 6, 2023 13:40 3m 40s
reshard params/opt_state post resume
Tests #1124: Commit adf90f4 pushed by fattorib
June 6, 2023 13:37 4m 17s pjit-tensor-parallel
June 6, 2023 13:37 4m 17s