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
805 workflow runs
805 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Fix test failure when shardy is not enabled. JAX Array API #826: Pull request #23979 opened by copybara-service bot
September 27, 2024 15:33 57s test_679601133
test_679601133
September 27, 2024 15:33 57s
Don't pass --nocheck_visibility to Bazel. JAX Array API #825: Pull request #23804 synchronize by hawkinsp
September 27, 2024 15:19 52s hawkinsp:vis
hawkinsp:vis
September 27, 2024 15:19 52s
Include the sdy MLIR dialect in jaxlib. JAX Array API #824: Pull request #23978 opened by copybara-service bot
September 27, 2024 15:03 59s test_679592303
test_679592303
September 27, 2024 15:03 59s
DOC: Improve docs for jax.numpy: arcsinh, arccosh and arctanh JAX Array API #823: Pull request #23977 synchronize by rajasekharporeddy
September 27, 2024 14:53 Action required rajasekharporeddy:testbranch1
rajasekharporeddy:testbranch1
September 27, 2024 14:53 Action required
DOC: Improve docs for jax.numpy: arcsinh, arccosh and arctanh JAX Array API #822: Pull request #23977 opened by rajasekharporeddy
September 27, 2024 14:48 Action required rajasekharporeddy:testbranch1
rajasekharporeddy:testbranch1
September 27, 2024 14:48 Action required
Test PR. JAX Array API #821: Pull request #23976 opened by copybara-service bot
September 27, 2024 14:25 53s test_679579784
test_679579784
September 27, 2024 14:25 53s
[Pallas/MGPU] Undo transforms on refs before giving them back to the users JAX Array API #820: Pull request #23974 opened by copybara-service bot
September 27, 2024 13:49 1m 7s test_679570949
test_679570949
September 27, 2024 13:49 1m 7s
Fix tests that ask for an accelerator but don't use it. JAX Array API #819: Pull request #23973 opened by copybara-service bot
September 27, 2024 13:27 55s test_679565592
test_679565592
September 27, 2024 13:27 55s
Replace disable_backends with enable_backends on jax_multiplatform_test. JAX Array API #818: Commit 26632fd pushed by copybara-service bot
September 27, 2024 13:15 52s main
main
September 27, 2024 13:15 52s
Replace disable_backends with enable_backends on jax_multiplatform_test. JAX Array API #817: Pull request #23953 synchronize by copybara-service bot
September 27, 2024 12:54 54s test_679219675
test_679219675
September 27, 2024 12:54 54s
Better documentation for jnp.choose JAX Array API #816: Pull request #23957 synchronize by jakevdp
September 27, 2024 12:50 57s jakevdp:choice-doc
jakevdp:choice-doc
September 27, 2024 12:50 57s
[Pallas TPU] Remove the skipTest for OpsExtraTest on TPU JAX Array API #815: Pull request #23967 synchronize by copybara-service bot
September 27, 2024 10:49 55s test_679493740
test_679493740
September 27, 2024 10:49 55s
[Pallas/MGPU] Skip output transfers when they don't depend on sequeni… JAX Array API #814: Commit 5740ab3 pushed by copybara-service bot
September 27, 2024 10:12 57s main
main
September 27, 2024 10:12 57s
[jax.distributed] Allow enabling grpc channel compression JAX Array API #813: Pull request #23969 opened by gspschmid
September 27, 2024 10:01 Action required gspschmid:gschmid/dist-compression
gspschmid:gschmid/dist-compression
September 27, 2024 10:01 Action required
[Pallas/MGPU] Skip output transfers when they don't depend on sequenital dims JAX Array API #812: Pull request #23945 synchronize by copybara-service bot
September 27, 2024 09:47 52s test_679129331
test_679129331
September 27, 2024 09:47 52s
Run Pallas Mosaic GPU tests on internal CI JAX Array API #811: Commit afaf8b8 pushed by copybara-service bot
September 27, 2024 09:43 51s main
main
September 27, 2024 09:43 51s
Run Pallas Mosaic GPU tests on internal CI JAX Array API #810: Pull request #23968 synchronize by copybara-service bot
September 27, 2024 09:29 1m 0s test_679494302
test_679494302
September 27, 2024 09:29 1m 0s
Fixed Pallas Mosaic GPU test following recent changes JAX Array API #809: Pull request #23955 synchronize by copybara-service bot
September 27, 2024 09:28 57s test_679261897
test_679261897
September 27, 2024 09:28 57s
Fixed Pallas Mosaic GPU test following recent changes JAX Array API #808: Commit 3ae4862 pushed by copybara-service bot
September 27, 2024 09:28 56s main
main
September 27, 2024 09:28 56s
[Pallas:TPU] Fix lowering of convert_element_type(int32) -> bool. JAX Array API #807: Commit ea86251 pushed by copybara-service bot
September 27, 2024 09:15 51s main
main
September 27, 2024 09:15 51s
Fixed Pallas Mosaic GPU test following recent changes JAX Array API #806: Pull request #23955 synchronize by copybara-service bot
September 27, 2024 09:15 1m 1s test_679261897
test_679261897
September 27, 2024 09:15 1m 1s
Removed unused imports in jax.experimental.mosaic.gpu.core JAX Array API #805: Commit ea6ee4d pushed by copybara-service bot
September 27, 2024 09:08 52s main
main
September 27, 2024 09:08 52s
[Pallas:TPU] Fix lowering of convert_element_type(int32) -> bool. JAX Array API #804: Pull request #23950 synchronize by copybara-service bot
September 27, 2024 08:58 55s test_679243567
test_679243567
September 27, 2024 08:58 55s
Removed unused imports in jax.experimental.mosaic.gpu.core JAX Array API #803: Pull request #23966 synchronize by copybara-service bot
September 27, 2024 08:56 54s test_679289036
test_679289036
September 27, 2024 08:56 54s
Run Pallas Mosaic GPU tests on internal CI JAX Array API #802: Pull request #23968 opened by copybara-service bot
September 27, 2024 08:55 1m 3s test_679494302
test_679494302
September 27, 2024 08:55 1m 3s