Skip to content

Actions: google/drjax

Actions

All workflows

Actions

Loading...
Loading

Showing runs from all workflows
72 workflow runs
72 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

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. Check for README file #2: Pull request #30 synchronize by copybara-service bot
7s cl/815172985
cl/815172985
7s
Ensure primitives propagate sharding annotations in abstract evaluation. Check for README file #1: Pull request #30 opened by copybara-service bot
6s cl/815172985
cl/815172985
6s
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
Replace jax.sharding.use_mesh with jax.set_mesh. jax.set_mesh can act as a global setter or a context manager. Test Project #54: Pull request #24 synchronize by copybara-service bot
32s cl/789513569
cl/789513569
32s
Replace jax.sharding.use_mesh with jax.set_mesh. jax.set_mesh can act as a global setter or a context manager. Test Project #53: Pull request #24 opened by copybara-service bot
32s cl/789513569
cl/789513569
32s
[JAX] Add a cache around HLO lowering rules. Test Project #52: Pull request #23 synchronize by copybara-service bot
51s cl/772600579
cl/772600579
51s
[JAX] Add a cache around HLO lowering rules. Test Project #51: Pull request #23 opened by copybara-service bot
45s cl/772600579
cl/772600579
45s
Change clients to placement in impls.py documentation. Test Project #50: Pull request #22 synchronize by copybara-service bot
54s cl/771440268
cl/771440268
54s
Change clients to placement in impls.py documentation. Test Project #49: Pull request #22 opened by copybara-service bot
50s cl/771440268
cl/771440268
50s
Opens up API hole enabling jax_privacy to disable ability to read jax abstract mesh in drjax. Test Project #48: Pull request #21 synchronize by copybara-service bot
5m 13s cl/765384845
cl/765384845
5m 13s