Skip to content

Actions: jax-ml/jax

Actions

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
ci-build.yaml
11,864 workflow run results
11,864 workflow run results

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Mosaic] Restore Python pipeline and add a CLI flag to run it. CI #48158: Pull request #20263 opened by copybara-service bot
March 15, 2024 00:54 5m 26s test_615959867
test_615959867
March 15, 2024 00:54 5m 26s
Clean up usage of jax.clear_backends CI #48157: Pull request #20262 opened by copybara-service bot
March 15, 2024 00:48 5m 38s test_615956199
test_615956199
March 15, 2024 00:48 5m 38s
[Pallas] Raise NotImplementedError for strided load/store in interpret mode. CI #48156: Pull request #20261 synchronize by copybara-service bot
March 15, 2024 00:37 5m 9s test_615955282
test_615955282
March 15, 2024 00:37 5m 9s
[Pallas] Raise NotImplementedError for strided load/store in interpret mode. CI #48155: Pull request #20261 synchronize by copybara-service bot
March 15, 2024 00:32 5m 23s test_615955282
test_615955282
March 15, 2024 00:32 5m 23s
[Pallas] Raise NotImplementedError for strided load/store in interpret mode. CI #48154: Pull request #20261 opened by copybara-service bot
March 15, 2024 00:30 1m 38s test_615955282
test_615955282
March 15, 2024 00:30 1m 38s
Add local_coordinator bool option CI #48152: Pull request #20260 opened by copybara-service bot
March 14, 2024 23:33 5m 23s test_615931461
test_615931461
March 14, 2024 23:33 5m 23s
[Pallas] Add stride in Pallas dynamic slice and support strided load/… CI #48151: Commit 2048e3c pushed by copybara-service bot
March 14, 2024 23:32 5m 21s main
main
March 14, 2024 23:32 5m 21s
[Pallas] Add stride in Pallas dynamic slice and support strided load/store. CI #48150: Pull request #20221 synchronize by copybara-service bot
March 14, 2024 23:12 5m 17s test_613710016
test_613710016
March 14, 2024 23:12 5m 17s
jax.clear_backends() is not doing what it is intended to do, users should try to avoid using it. CI #48149: Pull request #20219 synchronize by copybara-service bot
March 14, 2024 23:11 5m 48s test_615290629
test_615290629
March 14, 2024 23:11 5m 48s
jax.clear_backends() is not doing what it is intended to do, users … CI #48148: Commit 1cef1d9 pushed by copybara-service bot
March 14, 2024 23:11 5m 13s main
main
March 14, 2024 23:11 5m 13s
Fix jnp.digitize corner case with no bins CI #48147: Pull request #20258 synchronize by Gattocrucco
March 14, 2024 23:09 5m 14s Gattocrucco:digitize-fix-corner-case
Gattocrucco:digitize-fix-corner-case
March 14, 2024 23:09 5m 14s
[XLA:Mosaic] Support dynamic indices in strided load/store. CI #48146: Pull request #20220 synchronize by copybara-service bot
March 14, 2024 23:02 5m 7s test_615199548
test_615199548
March 14, 2024 23:02 5m 7s
[XLA:Mosaic] Support dynamic indices in strided load/store. CI #48145: Commit 7578e10 pushed by copybara-service bot
March 14, 2024 23:02 5m 21s main
main
March 14, 2024 23:02 5m 21s
jax.clear_backends() is not doing what it is intended to do, users should try to avoid using it. CI #48144: Pull request #20219 synchronize by copybara-service bot
March 14, 2024 22:56 5m 16s test_615290629
test_615290629
March 14, 2024 22:56 5m 16s
[XLA:Mosaic] Support dynamic indices in strided load/store. CI #48142: Pull request #20220 synchronize by copybara-service bot
March 14, 2024 22:17 5m 21s test_615199548
test_615199548
March 14, 2024 22:17 5m 21s
Guard host transfers inside pure_callbacks from deadlocking the TPU. CI #48141: Pull request #20259 opened by copybara-service bot
March 14, 2024 22:11 5m 19s test_615916557
test_615916557
March 14, 2024 22:11 5m 19s
[XLA:MHLO->HLO] Allow partially-set parameter tuple sharding to exist… CI #48139: Commit ac41032 pushed by copybara-service bot
March 14, 2024 22:10 5m 6s main
main
March 14, 2024 22:10 5m 6s
[XLA:MHLO->HLO] Allow partially-set parameter tuple sharding to exist by filling in the missing sharding elements with replicated sharding. (This is what is done for the missing shardings in the result tuple.) CI #48140: Pull request #20242 synchronize by copybara-service bot
March 14, 2024 22:10 5m 22s test_615287423
test_615287423
March 14, 2024 22:10 5m 22s
Fix jnp.digitize corner case with no bins CI #48138: Pull request #20258 opened by Gattocrucco
March 14, 2024 22:00 5m 14s Gattocrucco:digitize-fix-corner-case
Gattocrucco:digitize-fix-corner-case
March 14, 2024 22:00 5m 14s
[XLA:MHLO->HLO] Allow partially-set parameter tuple sharding to exist by filling in the missing sharding elements with replicated sharding. (This is what is done for the missing shardings in the result tuple.) CI #48137: Pull request #20242 synchronize by copybara-service bot
March 14, 2024 21:56 5m 14s test_615287423
test_615287423
March 14, 2024 21:56 5m 14s
[JAX cuDNN SDPA API] Add is_training and fix seqlen/head_dim checks CI #48136: Pull request #20257 synchronize by Cjkkkk
March 14, 2024 21:35 5m 44s Cjkkkk:sdpa_training
Cjkkkk:sdpa_training
March 14, 2024 21:35 5m 44s
Propagate effects errors to the results (only if effects are enabled). CI #48133: Commit 9a00721 pushed by copybara-service bot
March 14, 2024 20:32 5m 14s main
main
March 14, 2024 20:32 5m 14s
temporarily relax the cotangent dtype check introduced in #19009 CI #48131: Commit 6f38f27 pushed by copybara-service bot
March 14, 2024 20:22 5m 53s main
main
March 14, 2024 20:22 5m 53s
temporarily relax the cotangent dtype check introduced in #19009 CI #48132: Pull request #20255 synchronize by copybara-service bot
March 14, 2024 20:22 6m 1s test_615872948
test_615872948
March 14, 2024 20:22 6m 1s
temporarily relax the cotangent dtype check introduced in #19009 CI #48130: Pull request #20255 synchronize by copybara-service bot
March 14, 2024 19:58 5m 23s test_615872948
test_615872948
March 14, 2024 19:58 5m 23s