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

[FIX] Disable argument donation (again)
Docker Image CI #59: Commit b90f471 pushed by neel04
December 19, 2024 15:12 9m 45s dev
dev
December 19, 2024 15:12 9m 45s
Added more packages to the Dockerfile
Docker Image CI #58: Commit b623a9d pushed by neel04
November 25, 2024 13:22 8m 50s dev
dev
November 25, 2024 13:22 8m 50s
Removed TensorFlow dependency
Docker Image CI #57: Commit 43574c8 pushed by neel04
November 23, 2024 19:06 9m 41s dev
dev
November 23, 2024 19:06 9m 41s
Revamped profiling logic
Docker Image CI #56: Commit 8067e74 pushed by neel04
November 23, 2024 18:10 45m 5s dev
dev
November 23, 2024 18:10 45m 5s
Updated Dockerfile
Docker Image CI #55: Commit f55d7bc pushed by neel04
November 21, 2024 23:54 9m 2s dev
dev
November 21, 2024 23:54 9m 2s
Updated dockerfile with lm-eval
Docker Image CI #54: Commit 701cac5 pushed by neel04
October 23, 2024 11:55 8m 56s dev
dev
October 23, 2024 11:55 8m 56s
Updated dockerfile
Docker Image CI #53: Commit 98d371b pushed by neel04
October 23, 2024 11:51 8m 48s dev
dev
October 23, 2024 11:51 8m 48s
[FEAT] Added initial autosharding API
Docker Image CI #52: Commit a08abc5 pushed by neel04
September 3, 2024 01:44 8m 33s dev
dev
September 3, 2024 01:44 8m 33s
Removing scalax and updating docker image
Docker Image CI #51: Commit b7d0131 pushed by neel04
August 10, 2024 15:05 9m 23s dev
dev
August 10, 2024 15:05 9m 23s
chore: Update HF token and LR hyperparam schedule
Docker Image CI #50: Commit 507ba6a pushed by neel04
June 4, 2024 11:44 8m 13s dev
dev
June 4, 2024 11:44 8m 13s
Update HF token
Docker Image CI #49: Commit de5c6a4 pushed by neel04
June 3, 2024 12:01 9m 42s dev
dev
June 3, 2024 12:01 9m 42s
Attempt #2 at fixing optuna bug
Docker Image CI #48: Commit 2cf34f8 pushed by neel04
May 13, 2024 17:31 10m 26s dev
dev
May 13, 2024 17:31 10m 26s
Update Docker to accomodate jax rollback, fixing #21116
Docker Image CI #47: Commit c7e4033 pushed by neel04
May 10, 2024 00:27 8m 37s dev
dev
May 10, 2024 00:27 8m 37s
Added official JMP support for real mixed-precision
Docker Image CI #46: Commit a6d53e8 pushed by neel04
May 9, 2024 23:02 8m 58s dev
dev
May 9, 2024 23:02 8m 58s
Trigger docker rebuild
Docker Image CI #45: Commit 32e05d8 pushed by neel04
May 4, 2024 13:40 9m 6s dev
dev
May 4, 2024 13:40 9m 6s
added timeout handling for docker container removal
Docker Image CI #44: Commit f07a984 pushed by neel04
April 28, 2024 01:08 8m 39s dev
dev
April 28, 2024 01:08 8m 39s
Squash commit from dev to sync dev and main.
Docker Image CI #43: Commit 849c987 pushed by neel04
April 28, 2024 01:05 9m 10s dev
dev
April 28, 2024 01:05 9m 10s
added timeout handling for docker container removal
Docker Image CI #42: Commit f07a984 pushed by neel04
April 28, 2024 00:53 8m 13s dev
dev
April 28, 2024 00:53 8m 13s
Update data split to 40%
Docker Image CI #41: Commit cc0967d pushed by neel04
April 28, 2024 00:49 9m 41s dev
dev
April 28, 2024 00:49 9m 41s
Hyperparam search: 2 epochs over 20% data
Docker Image CI #40: Commit 2c4abe5 pushed by neel04
April 25, 2024 11:44 8m 39s dev
dev
April 25, 2024 11:44 8m 39s
Added EAI FLOPs calculation
Docker Image CI #39: Commit e7f1f96 pushed by neel04
April 14, 2024 23:55 8m 36s dev
dev
April 14, 2024 23:55 8m 36s
Convert numpy array to jax in the training loop
Docker Image CI #38: Commit 9f28c97 pushed by neel04
April 14, 2024 19:42 7m 38s dev
dev
April 14, 2024 19:42 7m 38s
EXP: Upgrade to a jax.lax.cond
Docker Image CI #37: Commit a572c47 pushed by neel04
April 14, 2024 10:44 7m 44s dev
dev
April 14, 2024 10:44 7m 44s
EXP: Use cross attention
Docker Image CI #36: Commit 1cb791d pushed by neel04
April 14, 2024 10:40 7m 48s dev
dev
April 14, 2024 10:40 7m 48s
FLOPs calculation + storing dataset on CPU during recomputation
Docker Image CI #35: Commit 61f5c4c pushed by neel04
April 13, 2024 22:13 8m 40s dev
dev
April 13, 2024 22:13 8m 40s