Skip to content

Actions: jax-ml/jax

Actions

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
ci-build.yaml
25,930 workflow runs
25,930 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Mosaic] Extend tpu matmulop to have dimension dims, modeled on jax's dot_general. Add support for batch dims via canonicalization to slice->matmul->concat CI #62625: Pull request #23994 opened by copybara-service bot
September 29, 2024 07:37 9m 16s test_680094115
test_680094115
September 29, 2024 07:37 9m 16s
Automated Code Change CI #62624: Pull request #23993 opened by copybara-service bot
September 29, 2024 04:45 9m 8s test_680115311
test_680115311
September 29, 2024 04:45 9m 8s
Update XLA dependency to use revision CI #62623: Commit 6790b90 pushed by copybara-service bot
September 28, 2024 17:54 9m 18s main
main
September 28, 2024 17:54 9m 18s
[Mosaic TPU] Force offset to 0 when inferring input has offset out of the first tile. CI #62622: Pull request #23987 synchronize by copybara-service bot
September 28, 2024 00:59 10m 45s test_679789606
test_679789606
September 28, 2024 00:59 10m 45s
[Mosaic TPU] Force offset to 0 when inferring input has offset out of the first tile. CI #62621: Pull request #23987 opened by copybara-service bot
September 28, 2024 00:59 1m 11s test_679789606
test_679789606
September 28, 2024 00:59 1m 11s
Merge pull request #23982 from dfm:ffi-call-effects CI #62620: Commit 15024ba pushed by copybara-service bot
September 28, 2024 00:15 9m 32s main
main
September 28, 2024 00:15 9m 32s
Add support for FFI calls with side effects via ffi_call CI #62619: Pull request #23982 synchronize by dfm
September 27, 2024 23:47 9m 20s dfm:ffi-call-effects
dfm:ffi-call-effects
September 27, 2024 23:47 9m 20s
[Mosaic TPU] Canonicalize arith.select's condition to vector if other types are vector. CI #62618: Pull request #23986 opened by copybara-service bot
September 27, 2024 23:21 10m 3s test_679764355
test_679764355
September 27, 2024 23:21 10m 3s
Add float8_e4m3 and float8_e3m4 types support CI #62617: Pull request #23585 synchronize by apivovarov
September 27, 2024 21:42 Action required apivovarov:float8_e4m3
apivovarov:float8_e4m3
September 27, 2024 21:42 Action required
Fix typo in Pallas TPU matmul doc. I think the logical layout of the input array is non-transposed, rather than transposed? CI #62616: Pull request #23985 opened by copybara-service bot
September 27, 2024 21:16 10m 22s test_679725301
test_679725301
September 27, 2024 21:16 10m 22s
jnp.mask_indices: add docs & tests CI #62615: Pull request #23984 opened by jakevdp
September 27, 2024 20:59 9m 9s jakevdp:mask-indices-doc
jakevdp:mask-indices-doc
September 27, 2024 20:59 9m 9s
Bump test tolerance on FFT test that started failing in CI after an XLA change. CI #62614: Pull request #23983 synchronize by copybara-service bot
September 27, 2024 20:50 10m 7s test_679706602
test_679706602
September 27, 2024 20:50 10m 7s
Bump test tolerance on FFT test that started failing in CI after an X… CI #62613: Commit 061f435 pushed by copybara-service bot
September 27, 2024 20:50 10m 42s main
main
September 27, 2024 20:50 10m 42s
Fix test failure when shardy is not enabled. CI #62612: Commit 366c823 pushed by copybara-service bot
September 27, 2024 20:42 7m 56s main
main
September 27, 2024 20:42 7m 56s
Bump test tolerance on FFT test that started failing in CI after an XLA change. CI #62611: Pull request #23983 synchronize by copybara-service bot
September 27, 2024 20:39 9m 27s test_679706602
test_679706602
September 27, 2024 20:39 9m 27s
Fix tests that ask for an accelerator but don't use it. CI #62610: Commit 5969e79 pushed by copybara-service bot
September 27, 2024 20:36 6m 18s main
main
September 27, 2024 20:36 6m 18s
Fix test failure when shardy is not enabled. CI #62609: Pull request #23979 synchronize by copybara-service bot
September 27, 2024 20:25 9m 27s test_679601133
test_679601133
September 27, 2024 20:25 9m 27s
Bump test tolerance on FFT test that started failing in CI after an XLA change. CI #62608: Pull request #23983 opened by copybara-service bot
September 27, 2024 20:22 9m 19s test_679706602
test_679706602
September 27, 2024 20:22 9m 19s
Fix tests that ask for an accelerator but don't use it. CI #62607: Pull request #23973 synchronize by copybara-service bot
September 27, 2024 20:19 9m 19s test_679565592
test_679565592
September 27, 2024 20:19 9m 19s
Activate FFI implementation of the QR decomposition. CI #62606: Pull request #23948 synchronize by copybara-service bot
September 27, 2024 19:25 9m 14s test_676825798
test_676825798
September 27, 2024 19:25 9m 14s
Activate FFI implementation of the QR decomposition. CI #62605: Pull request #23948 synchronize by copybara-service bot
September 27, 2024 19:08 8m 20s test_676825798
test_676825798
September 27, 2024 19:08 8m 20s
Add support for FFI calls with side effects via ffi_call CI #62604: Pull request #23982 opened by dfm
September 27, 2024 18:56 9m 32s dfm:ffi-call-effects
dfm:ffi-call-effects
September 27, 2024 18:56 9m 32s
Merge pull request #23957 from jakevdp:choice-doc CI #62603: Commit ff0a98a pushed by copybara-service bot
September 27, 2024 18:08 9m 40s main
main
September 27, 2024 18:08 9m 40s
Fix tests that ask for an accelerator but don't use it. CI #62602: Pull request #23973 synchronize by copybara-service bot
September 27, 2024 18:04 9m 47s test_679565592
test_679565592
September 27, 2024 18:04 9m 47s
Merge pull request #23981 from jakevdp:checkout-pin CI #62601: Commit 4e8a763 pushed by copybara-service bot
September 27, 2024 18:02 6m 56s main
main
September 27, 2024 18:02 6m 56s