Skip to content

Actions: google/drjax

Actions

Test Project

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
test_project.yaml
47 workflow runs
47 workflow runs

Filter by Event

⌥ + click/return to exclude

Filter by Status

⌥ + click/return to exclude

Filter by Branch

⌥ + click/return to exclude

Filter by Actor

⌥ + click/return to exclude
Update usage of deprecated jax.core APIs. Test Project #79: Pull request #36 synchronize by copybara-service bot
35s cl/893702651
cl/893702651
35s
Finalize deprecation of several jax.core APIs for JAX v0.10.0 Test Project #78: Pull request #35 synchronize by copybara-service bot
35s cl/893650230
cl/893650230
35s
Refactor drjax primitives to use jnp.broadcast_to and improve explicit sharding support. Test Project #77: Pull request #37 synchronize by copybara-service bot
33s cl/820412444
cl/820412444
33s
Refactor drjax primitives to use jnp.broadcast_to and improve explicit sharding support. Test Project #76: Pull request #37 opened by copybara-service bot
36s cl/820412444
cl/820412444
36s
Update usage of deprecated jax.core APIs. Test Project #75: Pull request #36 opened by copybara-service bot
37s cl/893702651
cl/893702651
37s
Finalize deprecation of several jax.core APIs for JAX v0.10.0 Test Project #74: Pull request #35 synchronize by copybara-service bot
36s cl/893650230
cl/893650230
36s
Finalize deprecation of several jax.core APIs for JAX v0.10.0 Test Project #73: Pull request #35 opened by copybara-service bot
37s cl/893650230
cl/893650230
37s
Ensure primitives propagate sharding annotations in abstract evaluation. Test Project #72: Pull request #34 synchronize by copybara-service bot
40s cl/893505609
cl/893505609
40s
Ensure primitives propagate sharding annotations in abstract evaluation. Test Project #71: Pull request #34 opened by copybara-service bot
41s cl/893505609
cl/893505609
41s
Ensure primitives propagate sharding annotations in abstract evaluation. Test Project #70: Pull request #33 opened by copybara-service bot
37s cl/821781134
cl/821781134
37s
Ensure primitives propagate sharding annotations in abstract evaluation. Test Project #69: Pull request #32 synchronize by copybara-service bot
45s cl/815172986
cl/815172986
45s
Ensure primitives propagate sharding annotations in abstract evaluation. Test Project #68: Pull request #32 opened by copybara-service bot
45s cl/815172986
cl/815172986
45s
Ensure primitives propagate sharding annotations in abstract evaluation. Test Project #67: Pull request #31 synchronize by copybara-service bot
35s cl/816789999
cl/816789999
35s
Ensure primitives propagate sharding annotations in abstract evaluation. Test Project #66: Pull request #31 opened by copybara-service bot
27s cl/816789999
cl/816789999
27s
Ensure primitives propagate sharding annotations in abstract evaluation. Test Project #65: Pull request #30 synchronize by copybara-service bot
35s cl/815172985
cl/815172985
35s
Ensure primitives propagate sharding annotations in abstract evaluation. Test Project #64: Pull request #30 opened by copybara-service bot
33s cl/815172985
cl/815172985
33s
Bug Fix: Tracers do not always shave .sharding, use jax.typeof to get an abstract value. Test Project #63: Pull request #29 synchronize by copybara-service bot
33s cl/815172984
cl/815172984
33s
Bug Fix: Tracers do not always shave .sharding, use jax.typeof to get an abstract value. Test Project #62: Pull request #29 opened by copybara-service bot
31s cl/815172984
cl/815172984
31s
Bug Fix: Tracers do not always shave .sharding, use jax.typeof to get an abstract value. Test Project #61: Pull request #28 opened by copybara-service bot
29s cl/815126678
cl/815126678
29s
Use Mesh properties for axis type checks. Test Project #60: Pull request #27 synchronize by copybara-service bot
34s cl/814228253
cl/814228253
34s
Use Mesh properties for axis type checks. Test Project #59: Pull request #27 opened by copybara-service bot
34s cl/814228253
cl/814228253
34s
Extend DrJax to be compatible with Explicit Sharding. Test Project #58: Pull request #26 synchronize by copybara-service bot
34s cl/812819695
cl/812819695
34s
Extend DrJax to be compatible with Explicit Sharding. Test Project #57: Pull request #26 opened by copybara-service bot
29s cl/812819695
cl/812819695
29s
[drjax] Avoid call to deprecated batching.moveaxis Test Project #56: Pull request #25 synchronize by copybara-service bot
28s cl/796450211
cl/796450211
28s
[drjax] Avoid call to deprecated batching.moveaxis Test Project #55: Pull request #25 opened by copybara-service bot
33s cl/796450211
cl/796450211
33s