Skip to content

Actions: jax-ml/jax

Actions

JAX Array API

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
jax-array-api.yml
7,236 workflow runs
7,236 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Mosaic:TPU] Vreg-slice-aligned offset changes with scratch retiling JAX Array API #7344: Pull request #25243 synchronize by copybara-service bot
December 20, 2024 20:23 3m 15s test_702557197
test_702557197
December 20, 2024 20:23 3m 15s
Fix attention backwards pass JAX Array API #7343: Pull request #25647 opened by Rifur13
December 20, 2024 20:19 3m 21s Rifur13:bwd_pass
Rifur13:bwd_pass
December 20, 2024 20:19 3m 21s
Add Github action workflows for running continuous tests with Pytest JAX Array API #7342: Pull request #25238 synchronize by copybara-service bot
December 20, 2024 20:18 3m 24s test_702497163
test_702497163
December 20, 2024 20:18 3m 24s
Add Github action workflows for running continuous tests with Pytest JAX Array API #7341: Pull request #25238 synchronize by copybara-service bot
December 20, 2024 20:15 3m 29s test_702497163
test_702497163
December 20, 2024 20:15 3m 29s
Migrate _mlir Python binding target to nanobind. JAX Array API #7340: Pull request #25619 synchronize by copybara-service bot
December 20, 2024 20:07 3m 18s test_708001140
test_708001140
December 20, 2024 20:07 3m 18s
Migrate _mlir Python binding target to nanobind. JAX Array API #7339: Commit 0ff3f14 pushed by copybara-service bot
December 20, 2024 20:07 3m 19s main
main
December 20, 2024 20:07 3m 19s
[Mosaic:TPU] Roll forward of cl/708011538 (expanded trunc support), minus changes in infer-vector-layout JAX Array API #7338: Pull request #25629 synchronize by copybara-service bot
December 20, 2024 20:00 3m 19s test_708184105
test_708184105
December 20, 2024 20:00 3m 19s
Add Github action workflows for running continuous tests with Pytest JAX Array API #7337: Pull request #25238 synchronize by copybara-service bot
December 20, 2024 19:58 3m 20s test_702497163
test_702497163
December 20, 2024 19:58 3m 20s
Migrate _mlir Python binding target to nanobind. JAX Array API #7336: Pull request #25619 synchronize by copybara-service bot
December 20, 2024 19:51 3m 25s test_708001140
test_708001140
December 20, 2024 19:51 3m 25s
Add int4/uint4 support to bitcast_convert_type JAX Array API #7335: Pull request #25646 opened by jakevdp
December 20, 2024 19:50 3m 17s jakevdp:int4-casting
jakevdp:int4-casting
December 20, 2024 19:50 3m 17s
[Mosaic TPU] Validate inserted layout in relayout-insertion pass. JAX Array API #7334: Pull request #25645 synchronize by copybara-service bot
December 20, 2024 19:42 3m 41s test_708378418
test_708378418
December 20, 2024 19:42 3m 41s
[Mosaic TPU] Validate inserted layout in relayout-insertion pass. JAX Array API #7333: Pull request #25645 opened by copybara-service bot
December 20, 2024 19:41 1m 30s test_708378418
test_708378418
December 20, 2024 19:41 1m 30s
Change the namespace name to avoid using export c++ keyword on namespace. JAX Array API #7332: Pull request #25644 synchronize by copybara-service bot
December 20, 2024 18:29 3m 44s test_708355966
test_708355966
December 20, 2024 18:29 3m 44s
Change the namespace name to avoid using export c++ keyword on namespace. JAX Array API #7331: Pull request #25644 synchronize by copybara-service bot
December 20, 2024 18:25 3m 42s test_708355966
test_708355966
December 20, 2024 18:25 3m 42s
Change the namespace name to avoid using export c++ keyword on namespace. JAX Array API #7330: Pull request #25644 synchronize by copybara-service bot
December 20, 2024 18:24 1m 14s test_708355966
test_708355966
December 20, 2024 18:24 1m 14s
Update XLA dependency to use revision JAX Array API #7329: Commit aa386f8 pushed by copybara-service bot
December 20, 2024 18:23 3m 24s main
main
December 20, 2024 18:23 3m 24s
Change the namespace name to avoid using export c++ keyword on namespace. JAX Array API #7328: Pull request #25644 opened by copybara-service bot
December 20, 2024 18:23 2m 3s test_708355966
test_708355966
December 20, 2024 18:23 2m 3s
Fix backwards attention test for larger head dims JAX Array API #7327: Pull request #25642 opened by Rifur13
December 20, 2024 18:01 3m 19s Rifur13:numerical_stability
Rifur13:numerical_stability
December 20, 2024 18:01 3m 19s
[Mosaic GPU] Add a lowering for simple async_load and async_store ops. JAX Array API #7326: Pull request #25635 synchronize by copybara-service bot
December 20, 2024 17:38 3m 22s test_707492485
test_707492485
December 20, 2024 17:38 3m 22s
[Mosaic GPU] Add a lowering for simple async_load and async_store… JAX Array API #7325: Commit dad23fe pushed by copybara-service bot
December 20, 2024 17:38 3m 18s main
main
December 20, 2024 17:38 3m 18s
Add Github action workflows for running continuous tests with Pytest JAX Array API #7324: Pull request #25238 synchronize by copybara-service bot
December 20, 2024 17:25 3m 30s test_702497163
test_702497163
December 20, 2024 17:25 3m 30s
[Mosaic GPU] Add a lowering for simple async_load and async_store ops. JAX Array API #7323: Pull request #25635 synchronize by copybara-service bot
December 20, 2024 17:23 3m 21s test_707492485
test_707492485
December 20, 2024 17:23 3m 21s
Merge pull request #25616 from jakevdp:abstractify JAX Array API #7322: Commit 01e8f88 pushed by copybara-service bot
December 20, 2024 17:09 3m 25s main
main
December 20, 2024 17:09 3m 25s
Add Github action workflows for running continuous tests with Pytest JAX Array API #7321: Pull request #25238 synchronize by copybara-service bot
December 20, 2024 17:03 3m 19s test_702497163
test_702497163
December 20, 2024 17:03 3m 19s
Add Github action workflows for running continuous tests with Pytest JAX Array API #7320: Pull request #25238 synchronize by copybara-service bot
December 20, 2024 16:55 3m 23s test_702497163
test_702497163
December 20, 2024 16:55 3m 23s