Skip to content

Actions: jax-ml/jax

Actions

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
ci-build.yaml
27,992 workflow runs
27,992 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[JAX] Optimize array shard reordering CI #70738: Pull request #26042 synchronize by copybara-service bot
January 23, 2025 03:29 10m 28s test_718482553
test_718482553
January 23, 2025 03:29 10m 28s
Support axis_index using a nested shard_map instead of iota with full to shard. CI #70737: Pull request #26049 synchronize by copybara-service bot
January 23, 2025 03:14 10m 30s test_718540794
test_718540794
January 23, 2025 03:14 10m 30s
Support axis_index using a nested shard_map instead of iota with full… CI #70736: Commit f3e27b6 pushed by copybara-service bot
January 23, 2025 03:14 10m 14s main
main
January 23, 2025 03:14 10m 14s
Add job that runs Bazel single accelerator and multi-accelerator CUDA tests (non-RBE) CI #70735: Pull request #25717 synchronize by copybara-service bot
January 23, 2025 01:51 9m 55s test_705238265
test_705238265
January 23, 2025 01:51 9m 55s
Add job that runs Bazel single accelerator and multi-accelerator CUDA… CI #70734: Commit 9aad6a6 pushed by copybara-service bot
January 23, 2025 01:51 11m 37s main
main
January 23, 2025 01:51 11m 37s
Add job that runs Bazel single accelerator and multi-accelerator CUDA tests (non-RBE) CI #70733: Pull request #25717 synchronize by copybara-service bot
January 23, 2025 01:38 10m 17s test_705238265
test_705238265
January 23, 2025 01:38 10m 17s
Do not check device_count constraint for some users. CI #70732: Pull request #26025 synchronize by copybara-service bot
January 23, 2025 00:59 10m 29s test_716346724
test_716346724
January 23, 2025 00:59 10m 29s
[sharding_in_types] Make vmap work with shard_map + pallas CI #70731: Commit 704b2e5 pushed by copybara-service bot
January 23, 2025 00:48 9m 50s main
main
January 23, 2025 00:48 9m 50s
[sharding_in_types] Make vmap work with shard_map + pallas CI #70730: Pull request #26047 synchronize by copybara-service bot
January 23, 2025 00:47 10m 14s test_718511907
test_718511907
January 23, 2025 00:47 10m 14s
[JAX] Optimize array shard reordering CI #70729: Pull request #26042 synchronize by copybara-service bot
January 23, 2025 00:46 10m 33s test_718482553
test_718482553
January 23, 2025 00:46 10m 33s
[JAX] Optimize array shard reordering CI #70728: Pull request #26042 synchronize by copybara-service bot
January 23, 2025 00:45 1m 22s test_718482553
test_718482553
January 23, 2025 00:45 1m 22s
Do not check device_count constraint for some users. CI #70727: Pull request #26025 synchronize by copybara-service bot
January 23, 2025 00:44 7m 33s test_716346724
test_716346724
January 23, 2025 00:44 7m 33s
Do not check device_count constraint for some users. CI #70726: Pull request #26025 synchronize by copybara-service bot
January 23, 2025 00:33 7m 13s test_716346724
test_716346724
January 23, 2025 00:33 7m 13s
Support axis_index using a nested shard_map instead of iota with full to shard. CI #70725: Pull request #26049 synchronize by copybara-service bot
January 23, 2025 00:33 10m 10s test_718540794
test_718540794
January 23, 2025 00:33 10m 10s
Do not check device_count constraint for some users. CI #70724: Pull request #26025 synchronize by copybara-service bot
January 23, 2025 00:25 6m 45s test_716346724
test_716346724
January 23, 2025 00:25 6m 45s
[JAX] Optimize array shard reordering CI #70723: Pull request #26042 synchronize by copybara-service bot
January 23, 2025 00:25 10m 31s test_718482553
test_718482553
January 23, 2025 00:25 10m 31s
Speed up name stack printing. CI #70722: Pull request #26036 synchronize by copybara-service bot
January 23, 2025 00:24 10m 29s test_718383947
test_718383947
January 23, 2025 00:24 10m 29s
Speed up name stack printing. CI #70721: Commit cd51e9d pushed by copybara-service bot
January 23, 2025 00:24 10m 31s main
main
January 23, 2025 00:24 10m 31s
[JAX] Optimize array shard reordering CI #70720: Pull request #26042 synchronize by copybara-service bot
January 23, 2025 00:20 5m 4s test_718482553
test_718482553
January 23, 2025 00:20 5m 4s
Initialize BlockSizes from attention.BlockSizes only on non-Windows platforms CI #70719: Pull request #26052 opened by copybara-service bot
January 23, 2025 00:12 9m 24s test_718563488
test_718563488
January 23, 2025 00:12 9m 24s
Merge pull request #26050 from jakevdp:effver-adoption CI #70718: Commit 0fccabc pushed by copybara-service bot
January 23, 2025 00:01 10m 10s main
main
January 23, 2025 00:01 10m 10s
[sharding_in_types] Make vmap work with shard_map + pallas CI #70717: Pull request #26047 synchronize by copybara-service bot
January 22, 2025 23:58 10m 27s test_718511907
test_718511907
January 22, 2025 23:58 10m 27s
Update debugging Pallas g3doc to remove text about scalar printing restriction CI #70716: Pull request #26048 synchronize by copybara-service bot
January 22, 2025 23:55 10m 25s test_718537247
test_718537247
January 22, 2025 23:55 10m 25s
Update debugging Pallas g3doc to remove text about scalar printing re… CI #70715: Commit 64e9b07 pushed by copybara-service bot
January 22, 2025 23:55 6m 39s main
main
January 22, 2025 23:55 6m 39s
Adds a minimal but viable implementation of string arrays (with numpy.dtypes.StringDType) in JAX. Currently this only supports making of a string array by means of either jax.numpy.asarray or jax.device_put and reading it back with jax.device_get. CI #70714: Pull request #25918 synchronize by copybara-service bot
January 22, 2025 23:51 10m 50s test_716042460
test_716042460
January 22, 2025 23:51 10m 50s