Skip to content

Actions: jax-ml/jax

CI - Bazel CPU tests (RBE)

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
4,330 workflow runs
4,330 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Pallas] Add option for async DMAs in the new TPU interpret mode
CI - Bazel CPU tests (RBE) #4403: Pull request #26650 opened by copybara-service bot
February 20, 2025 23:41 5m 13s test_729245019
February 20, 2025 23:41 5m 13s
roofline: Handle ClosedJaxpr instances.
CI - Bazel CPU tests (RBE) #4402: Pull request #26648 synchronize by copybara-service bot
February 20, 2025 23:20 3m 34s test_729230803
February 20, 2025 23:20 3m 34s
jax.numpy ndim/shape/size: deprecate non-array input
CI - Bazel CPU tests (RBE) #4401: Pull request #26641 synchronize by jakevdp
February 20, 2025 23:19 4m 29s jakevdp:jnp-ndim
February 20, 2025 23:19 4m 29s
Merge pull request #26646 from jakevdp:fix-rtd
CI - Bazel CPU tests (RBE) #4400: Commit bbc4fa7 pushed by copybara-service bot
February 20, 2025 23:17 3m 32s main
February 20, 2025 23:17 3m 32s
Internal: improved type annotations for lu.WrappedFun
CI - Bazel CPU tests (RBE) #4399: Pull request #26649 synchronize by jakevdp
February 20, 2025 23:12 5m 30s jakevdp:wrapped-fun
February 20, 2025 23:12 5m 30s
roofline: Handle ClosedJaxpr instances.
CI - Bazel CPU tests (RBE) #4398: Pull request #26648 synchronize by copybara-service bot
February 20, 2025 23:12 6m 8s test_729230803
February 20, 2025 23:12 6m 8s
[pallas:mosaic_gpu] Added support for binary/comparison ops with WG s…
CI - Bazel CPU tests (RBE) #4397: Commit 7438976 pushed by copybara-service bot
February 20, 2025 23:06 8m 13s main
February 20, 2025 23:06 8m 13s
Internal: improved type annotations for lu.WrappedFun
CI - Bazel CPU tests (RBE) #4396: Pull request #26649 synchronize by jakevdp
February 20, 2025 23:06 6m 42s jakevdp:wrapped-fun
February 20, 2025 23:06 6m 42s
Internal: improved type annotations for lu.WrappedFun
CI - Bazel CPU tests (RBE) #4395: Pull request #26649 opened by jakevdp
February 20, 2025 22:38 5m 20s jakevdp:wrapped-fun
February 20, 2025 22:38 5m 20s
[pallas:mosaic_gpu] Added support for binary/comparison ops with WG semantics
CI - Bazel CPU tests (RBE) #4394: Pull request #26634 synchronize by copybara-service bot
February 20, 2025 22:32 2m 49s test_729078275
February 20, 2025 22:32 2m 49s
Plumb layout through the creation of PjRtArrays.
CI - Bazel CPU tests (RBE) #4393: Pull request #26623 synchronize by copybara-service bot
February 20, 2025 22:27 4m 26s test_724096815
February 20, 2025 22:27 4m 26s
[pallas:mosaic_gpu] Added support for binary/comparison ops with WG semantics
CI - Bazel CPU tests (RBE) #4392: Pull request #26634 synchronize by copybara-service bot
February 20, 2025 22:20 5m 2s test_729078275
February 20, 2025 22:20 5m 2s
Use new ML Build CUDA images
CI - Bazel CPU tests (RBE) #4391: Pull request #26644 synchronize by copybara-service bot
February 20, 2025 22:01 7m 41s test_729207975
February 20, 2025 22:01 7m 41s
roofline: Add support for min_p, max_p, reduce_sum_p.
CI - Bazel CPU tests (RBE) #4390: Pull request #26643 synchronize by copybara-service bot
February 20, 2025 21:59 6m 34s test_729198549
February 20, 2025 21:59 6m 34s
roofline: Handle ClosedJaxpr instances.
CI - Bazel CPU tests (RBE) #4389: Pull request #26648 opened by copybara-service bot
February 20, 2025 21:33 4m 36s test_729230803
February 20, 2025 21:33 4m 36s
Fix head comment: was referring to nonexistent parameters.
CI - Bazel CPU tests (RBE) #4388: Pull request #26647 synchronize by copybara-service bot
February 20, 2025 21:29 5m 1s test_729215847
February 20, 2025 21:29 5m 1s
Fix head comment: was referring to nonexistent parameters.
CI - Bazel CPU tests (RBE) #4387: Commit 08de012 pushed by copybara-service bot
February 20, 2025 21:29 3m 40s main
February 20, 2025 21:29 3m 40s
CI: skip execution of convolutions.ipynb
CI - Bazel CPU tests (RBE) #4386: Pull request #26646 synchronize by jakevdp
February 20, 2025 21:18 2m 47s jakevdp:fix-rtd
February 20, 2025 21:18 2m 47s
roofline: Add support for min_p, max_p, reduce_sum_p.
CI - Bazel CPU tests (RBE) #4385: Pull request #26643 synchronize by copybara-service bot
February 20, 2025 21:09 4m 44s test_729198549
February 20, 2025 21:09 4m 44s
Fix head comment: was referring to nonexistent parameters.
CI - Bazel CPU tests (RBE) #4384: Pull request #26647 synchronize by copybara-service bot
February 20, 2025 21:01 3m 47s test_729215847
February 20, 2025 21:01 3m 47s
Support Scaled matmul and scaled dot_general in jax.nn
CI - Bazel CPU tests (RBE) #4383: Pull request #26345 synchronize by wenscarl
February 20, 2025 20:52 4m 50s wenscarl:scaled_matmul
February 20, 2025 20:52 4m 50s
Fix head comment: was referring to nonexistent parameters.
CI - Bazel CPU tests (RBE) #4382: Pull request #26647 opened by copybara-service bot
February 20, 2025 20:52 5m 14s test_729215847
February 20, 2025 20:52 5m 14s
[JAX] Implement an initial object API for colocated Python
CI - Bazel CPU tests (RBE) #4381: Pull request #26626 synchronize by copybara-service bot
February 20, 2025 20:52 4m 16s test_720733692
February 20, 2025 20:52 4m 16s
CI: skip execution of convolutions.ipynb
CI - Bazel CPU tests (RBE) #4380: Pull request #26646 opened by jakevdp
February 20, 2025 20:41 2m 44s jakevdp:fix-rtd
February 20, 2025 20:41 2m 44s
jax.numpy ndim/shape/size: deprecate non-array input
CI - Bazel CPU tests (RBE) #4379: Pull request #26641 synchronize by jakevdp
February 20, 2025 20:31 6m 0s jakevdp:jnp-ndim
February 20, 2025 20:31 6m 0s