Skip to content

Actions: jax-ml/jax

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
25,924 workflow runs
25,924 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Pallas/MGPU] Implement block spec evaluation correctly
CI #62499: Pull request #23941 synchronize by copybara-service bot
September 26, 2024 16:04 9m 24s test_679102766
September 26, 2024 16:04 9m 24s
#sdy Add CPU targets in JAX.
CI #62498: Pull request #23901 synchronize by copybara-service bot
September 26, 2024 16:02 9m 18s test_678626718
September 26, 2024 16:02 9m 18s
#sdy add JAX Shardy support for shard_map.
CI #62497: Commit e62a50c pushed by copybara-service bot
September 26, 2024 15:45 10m 43s main
September 26, 2024 15:45 10m 43s
Add jax.errors.JaxRuntimeError as a public alias for the XlaRuntimeError class.
CI #62496: Pull request #23943 synchronize by copybara-service bot
September 26, 2024 15:39 10m 23s test_679136569
September 26, 2024 15:39 10m 23s
September 26, 2024 15:39 6m 36s
Activate FFI implementation of the QR decomposition.
CI #62494: Pull request #23948 opened by copybara-service bot
September 26, 2024 15:37 7m 7s test_676825798
September 26, 2024 15:37 7m 7s
Added support for lax.cond_p to Pallas Mosaic GPU lowering
CI #62491: Commit 5cef547 pushed by copybara-service bot
September 26, 2024 15:20 9m 42s main
September 26, 2024 15:20 9m 42s
#sdy add JAX Shardy support for shard_map.
CI #62490: Pull request #23799 synchronize by copybara-service bot
September 26, 2024 15:16 10m 49s test_672993069
September 26, 2024 15:16 10m 49s
Added support for lax.cond_p to Pallas Mosaic GPU lowering
CI #62489: Pull request #23920 synchronize by copybara-service bot
September 26, 2024 15:01 10m 41s test_676133222
September 26, 2024 15:01 10m 41s
Add jax.errors.JaxRuntimeError as a public alias for the XlaRuntimeError class.
CI #62488: Pull request #23943 synchronize by copybara-service bot
September 26, 2024 14:59 10m 47s test_679136569
September 26, 2024 14:59 10m 47s
[Pallas/MGPU] Fix a race in the pipelining code
CI #62487: Commit 0a66e2d pushed by copybara-service bot
September 26, 2024 14:58 9m 37s main
September 26, 2024 14:58 9m 37s
[Pallas/MGPU] Fix a race in the pipelining code
CI #62485: Pull request #23939 synchronize by copybara-service bot
September 26, 2024 14:46 10m 0s test_679102765
September 26, 2024 14:46 10m 0s
Add jax.errors.JaxRuntimeError as a public alias for the XlaRuntimeError class.
CI #62484: Pull request #23943 synchronize by copybara-service bot
September 26, 2024 14:45 10m 31s test_679136569
September 26, 2024 14:45 10m 31s
Deprecate jax.lib.xla_client.Device.
CI #62483: Pull request #23944 opened by copybara-service bot
September 26, 2024 14:40 10m 26s test_679143844
September 26, 2024 14:40 10m 26s
[Pallas/Mosaic GPU] Implement a more comprehensive matmul kernel to s…
CI #62481: Commit 8599dbc pushed by copybara-service bot
September 26, 2024 14:40 10m 38s main
September 26, 2024 14:40 10m 38s
Merge pull request #23927 from jakevdp:pad-doc
CI #62480: Commit c07652f pushed by copybara-service bot
September 26, 2024 14:37 3m 35s main
September 26, 2024 14:37 3m 35s