Skip to content

Actions: jax-ml/jax

Actions

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

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

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[XLA:Mosaic][Pallas] Enable vector.ExtractOp for non-zero indices. CI #62524: Pull request #23954 synchronize by copybara-service bot
September 26, 2024 19:59 9m 43s test_665053223
test_665053223
September 26, 2024 19:59 9m 43s
[XLA:Mosaic][Pallas] Enable vector.ExtractOp for non-zero indices. CI #62523: Pull request #23954 opened by copybara-service bot
September 26, 2024 19:55 3m 59s test_665053223
test_665053223
September 26, 2024 19:55 3m 59s
Rewrite mosaic concat to support operand shapes that do not align with native shapes, Expand tests to cover multi operand, batch dim concat, etc. CI #62522: Pull request #23714 synchronize by copybara-service bot
September 26, 2024 19:33 9m 19s test_675835891
test_675835891
September 26, 2024 19:33 9m 19s
Replace disable_backends with enable_backends on jax_multiplatform_test. CI #62521: Pull request #23953 opened by copybara-service bot
September 26, 2024 19:31 9m 49s test_679219675
test_679219675
September 26, 2024 19:31 9m 49s
Add support for passing array attributes via ffi_call and an example demonstrating the different attribute APIs supported by the FFI CI #62520: Pull request #23951 synchronize by dfm
September 26, 2024 19:27 10m 11s dfm:ffi-examples-attrs
dfm:ffi-examples-attrs
September 26, 2024 19:27 10m 11s
Add support for passing array attributes via ffi_call and an example demonstrating the different attribute APIs supported by the FFI CI #62519: Pull request #23951 opened by dfm
September 26, 2024 19:22 5m 38s dfm:ffi-examples-attrs
dfm:ffi-examples-attrs
September 26, 2024 19:22 5m 38s
[Pallas:TPU] Fix lowering of convert_element_type(int32) -> bool. CI #62518: Pull request #23950 opened by copybara-service bot
September 26, 2024 19:11 9m 19s test_679243567
test_679243567
September 26, 2024 19:11 9m 19s
DOC: Improve docs for jax.numpy: sinh, cosh and tanh CI #62517: Pull request #23949 opened by rajasekharporeddy
September 26, 2024 17:59 10m 5s rajasekharporeddy:testbranch1
rajasekharporeddy:testbranch1
September 26, 2024 17:59 10m 5s
[Mosaic TPU] Support bitcast with non-native tiling and reset offsets when not falling within the first tile. CI #62516: Pull request #23934 synchronize by copybara-service bot
September 26, 2024 17:40 9m 37s test_678923806
test_678923806
September 26, 2024 17:40 9m 37s
Merge pull request #23940 from jakevdp:jacobian-doc CI #62515: Commit 6f7ad64 pushed by copybara-service bot
September 26, 2024 17:34 10m 54s main
main
September 26, 2024 17:34 10m 54s
Merge pull request #23921 from rajasekharporeddy:testbranch4 CI #62514: Commit 96cf2b8 pushed by copybara-service bot
September 26, 2024 17:32 2m 1s main
main
September 26, 2024 17:32 2m 1s
Introduce a pl.actual_size(ref_idx) utility to make masking in pallas easier. CI #62513: Pull request #23832 synchronize by copybara-service bot
September 26, 2024 17:28 9m 11s test_677023684
test_677023684
September 26, 2024 17:28 9m 11s
[WIP] jax pytree data serialization CI #62512: Pull request #23856 synchronize by copybara-service bot
September 26, 2024 17:25 10m 35s test_668858634
test_668858634
September 26, 2024 17:25 10m 35s
Deprecate jax.lib.xla_client.Device. CI #62511: Pull request #23944 synchronize by copybara-service bot
September 26, 2024 17:17 9m 29s test_679143844
test_679143844
September 26, 2024 17:17 9m 29s
Deprecate jax.lib.xla_client.Device. CI #62510: Commit 0e082f9 pushed by copybara-service bot
September 26, 2024 17:17 11m 22s main
main
September 26, 2024 17:17 11m 22s
Update XLA dependency to use revision CI #62509: Commit 140a8c7 pushed by copybara-service bot
September 26, 2024 17:13 3m 45s main
main
September 26, 2024 17:13 3m 45s
Deprecate jax.lib.xla_client.Device. CI #62508: Pull request #23944 synchronize by copybara-service bot
September 26, 2024 16:58 10m 7s test_679143844
test_679143844
September 26, 2024 16:58 10m 7s
[Pallas/MGPU] Skip outgoing TMA when the output is being revisited CI #62507: Commit dd2ee8c pushed by copybara-service bot
September 26, 2024 16:54 10m 10s main
main
September 26, 2024 16:54 10m 10s
[Pallas/MGPU] Skip outgoing TMA when the output is being revisited CI #62506: Pull request #23942 synchronize by copybara-service bot
September 26, 2024 16:34 10m 25s test_679111363
test_679111363
September 26, 2024 16:34 10m 25s
[WIP] jax pytree data serialization CI #62505: Pull request #23856 synchronize by copybara-service bot
September 26, 2024 16:31 9m 34s test_668858634
test_668858634
September 26, 2024 16:31 9m 34s
[Pallas/MGPU] Implement block spec evaluation correctly CI #62504: Commit 076287f pushed by copybara-service bot
September 26, 2024 16:15 9m 16s main
main
September 26, 2024 16:15 9m 16s
#sdy Add CPU targets in JAX. CI #62503: Pull request #23901 synchronize by copybara-service bot
September 26, 2024 16:13 11m 46s test_678626718
test_678626718
September 26, 2024 16:13 11m 46s
#sdy Add CPU targets in JAX. CI #62502: Commit a3284bd pushed by copybara-service bot
September 26, 2024 16:13 1m 56s main
main
September 26, 2024 16:13 1m 56s
Raise ValueError when axis1==axis2 for jnp.trace CI #62501: Pull request #23921 synchronize by rajasekharporeddy
September 26, 2024 16:08 9m 21s rajasekharporeddy:testbranch4
rajasekharporeddy:testbranch4
September 26, 2024 16:08 9m 21s
Raise ValueError when axis1==axis2 for jnp.trace CI #62500: Pull request #23921 synchronize by rajasekharporeddy
September 26, 2024 16:07 Action required rajasekharporeddy:testbranch4
rajasekharporeddy:testbranch4
September 26, 2024 16:07 Action required