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

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[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
[Pallas TPU] Remove the skipTest for OpsExtraTest on TPU JAX Array API #801: Pull request #23967 opened by copybara-service bot
September 27, 2024 08:54 53s test_679493740
test_679493740
September 27, 2024 08:54 53s
Removed unused imports in jax.experimental.mosaic.gpu.core JAX Array API #800: Pull request #23966 opened by copybara-service bot
September 27, 2024 08:49 47s test_679289036
test_679289036
September 27, 2024 08:49 47s
[Pallas TPU] Add a note in the Pallas Quickstart documentation about … JAX Array API #799: Commit ab4590c pushed by copybara-service bot
September 27, 2024 08:33 58s main
main
September 27, 2024 08:33 58s
[Pallas TPU] Add a note in the Pallas Quickstart documentation about the instructions of running the existing example on TPU JAX Array API #798: Pull request #23889 synchronize by copybara-service bot
September 27, 2024 08:08 50s test_678420788
test_678420788
September 27, 2024 08:08 50s
Enable weight offloading tests that are supported on GPUs now JAX Array API #797: Pull request #23965 opened by zhenying-liu
September 27, 2024 06:46 56s zhenying-liu:weight-offloading-test
zhenying-liu:weight-offloading-test
September 27, 2024 06:46 56s
Update device_memory_profiling.md JAX Array API #796: Pull request #23964 opened by garymm
September 27, 2024 05:10 Action required garymm:patch-2
garymm:patch-2
September 27, 2024 05:10 Action required
Add float8_e4m3 and float8_e3m4 types support JAX Array API #795: Pull request #23585 synchronize by apivovarov
September 27, 2024 03:41 1m 1s apivovarov:float8_e4m3
apivovarov:float8_e4m3
September 27, 2024 03:41 1m 1s
Add float8_e4m3 and float8_e3m4 types support JAX Array API #794: Pull request #23585 synchronize by apivovarov
September 27, 2024 03:39 Action required apivovarov:float8_e4m3
apivovarov:float8_e4m3
September 27, 2024 03:39 Action required
Add float8_e4m3 and float8_e3m4 types support JAX Array API #793: Pull request #23585 synchronize by apivovarov
September 27, 2024 03:38 Action required apivovarov:float8_e4m3
apivovarov:float8_e4m3
September 27, 2024 03:38 Action required
Adding JAX_LOGGING_LEVEL configuration option JAX Array API #792: Pull request #23468 synchronize by rdyro
September 27, 2024 01:58 58s rdyro:rdyro-add-logging-env
rdyro:rdyro-add-logging-env
September 27, 2024 01:58 58s
Merge pull request #23853 from zhenying-liu:remat-scan JAX Array API #791: Commit 5a1549c pushed by copybara-service bot
September 27, 2024 01:12 52s main
main
September 27, 2024 01:12 52s