Skip to content

Actions: neel04/ReAct_Jax

Actions

All workflows

Actions

Loading...
Loading

Showing runs from all workflows
59 workflow runs
59 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

WTF is wrong with git
Docker Image CI #34: Pull request #13 opened by neel04
April 11, 2024 23:48 8m 14s dev
dev
April 11, 2024 23:48 8m 14s
Fixed LR scheduling=
Docker Image CI #33: Commit 9f67154 pushed by neel04
April 11, 2024 20:41 9m 23s dev
dev
April 11, 2024 20:41 9m 23s
Putting the numpificaiton of data in the trainloader
Docker Image CI #32: Commit b3d7540 pushed by neel04
April 11, 2024 15:59 7m 51s dev
dev
April 11, 2024 15:59 7m 51s
Sharding compute_metrics
Docker Image CI #31: Commit 7e839ee pushed by neel04
April 11, 2024 13:30 8m 3s dev
dev
April 11, 2024 13:30 8m 3s
Fully removed TP code
Docker Image CI #30: Commit 6c9385e pushed by neel04
April 10, 2024 22:56 8m 28s dev
dev
April 10, 2024 22:56 8m 28s
Don't donate model
Docker Image CI #29: Commit f185723 pushed by neel04
April 10, 2024 21:22 9m 1s dev
dev
April 10, 2024 21:22 9m 1s
reduce comms overhead
Docker Image CI #28: Commit 3a9b7b3 pushed by neel04
April 10, 2024 00:03 8m 17s dev
dev
April 10, 2024 00:03 8m 17s
Turn off TP. OnlyDDP
Docker Image CI #27: Commit f3c761d pushed by neel04
April 9, 2024 22:53 8m 5s dev
dev
April 9, 2024 22:53 8m 5s
precasting the data to jax arrays
Docker Image CI #26: Commit 66bde33 pushed by neel04
April 8, 2024 00:04 8m 23s dev
dev
April 8, 2024 00:04 8m 23s
report devices for tuning hyperparams
Docker Image CI #25: Commit 894ffc2 pushed by neel04
April 7, 2024 23:34 8m 21s dev
dev
April 7, 2024 23:34 8m 21s
Support for DDP + TP
Docker Image CI #24: Commit 96efca6 pushed by neel04
April 7, 2024 01:11 8m 0s dev
dev
April 7, 2024 01:11 8m 0s
FSDP -> DDP. Dataset length autodetection
Docker Image CI #23: Commit 24accfd pushed by neel04
April 5, 2024 23:40 7m 58s dev
dev
April 5, 2024 23:40 7m 58s
all gather sharded model before serializing it
Docker Image CI #22: Commit 7f418f2 pushed by neel04
April 5, 2024 22:25 8m 20s dev
dev
April 5, 2024 22:25 8m 20s
cast to jnp array
Docker Image CI #21: Commit 7822b11 pushed by neel04
April 5, 2024 20:33 8m 13s dev
dev
April 5, 2024 20:33 8m 13s
fixed FSDP not doing DP on compute_metrics
Docker Image CI #20: Commit b86b75d pushed by neel04
April 5, 2024 20:02 8m 24s dev
dev
April 5, 2024 20:02 8m 24s
disabling DP in metrics computation
Docker Image CI #19: Commit e31cf49 pushed by neel04
April 5, 2024 13:21 8m 6s dev
dev
April 5, 2024 13:21 8m 6s
sharding metrics computation to be fsdp
Docker Image CI #18: Commit 9f0ab4d pushed by neel04
April 5, 2024 13:13 16m 10s dev
dev
April 5, 2024 13:13 16m 10s
Initial FSDP code
Docker Image CI #17: Commit b81e458 pushed by neel04
April 5, 2024 09:37 7m 45s dev
dev
April 5, 2024 09:37 7m 45s
reduced sweep dataset slice
Docker Image CI #16: Commit 2b942b0 pushed by neel04
April 4, 2024 14:17 18m 8s dev
dev
April 4, 2024 14:17 18m 8s
speedup all_gather
Docker Image CI #15: Commit 97e8ecd pushed by neel04
April 3, 2024 13:24 10m 32s dev
dev
April 3, 2024 13:24 10m 32s
put the LN back
Docker Image CI #14: Commit 879dc8f pushed by neel04
April 2, 2024 13:08 8m 7s dev
dev
April 2, 2024 13:08 8m 7s
removed LN and fixed sampling
Docker Image CI #13: Commit 5c180ff pushed by neel04
April 2, 2024 10:34 15m 35s dev
dev
April 2, 2024 10:34 15m 35s
double the sweeps data
Docker Image CI #12: Commit 8eecef2 pushed by neel04
April 1, 2024 20:12 8m 10s dev
dev
April 1, 2024 20:12 8m 10s
Docker Image CI
Docker Image CI #11: by neel04
April 1, 2024 19:45 10m 21s dev
dev
April 1, 2024 19:45 10m 21s
use vmap to reduce compilation time for react
Docker Image CI #10: Commit 569d37d pushed by neel04
March 30, 2024 18:37 8m 48s dev
dev
March 30, 2024 18:37 8m 48s