Skip to content

Actions: jax-ml/jax

Actions

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
ci-build.yaml
31,915 workflow runs
31,915 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Pallas TPU] Avoid using SMEM tensors in pipeline loop to track buffer slots CI #83511: Pull request #29204 synchronize by copybara-service bot
June 3, 2025 18:45 11m 59s test_765952139
test_765952139
June 3, 2025 18:45 11m 59s
fix missing input checks in jax.nn.functions CI #83510: Pull request #29206 synchronize by DanisNone
June 3, 2025 18:42 Action required DanisNone:main
DanisNone:main
June 3, 2025 18:42 Action required
Prototype of cross-host device transfers in IFRT-PJRT. CI #83509: Pull request #28867 synchronize by copybara-service bot
June 3, 2025 18:41 10m 31s test_761220528
test_761220528
June 3, 2025 18:41 10m 31s
[Pallas TPU] Avoid using SMEM tensors in pipeline loop to track buffer slots CI #83508: Pull request #29204 synchronize by copybara-service bot
June 3, 2025 18:40 4m 34s test_765952139
test_765952139
June 3, 2025 18:40 4m 34s
fix missing input checks in jax.nn.functions CI #83507: Pull request #29206 synchronize by DanisNone
June 3, 2025 18:40 Action required DanisNone:main
DanisNone:main
June 3, 2025 18:40 Action required
[Mosaic GPU] Add BUILD rules for blackwell matmul kernel CI #83506: Commit 554cc01 pushed by copybara-service bot
June 3, 2025 18:25 7m 10s main
main
June 3, 2025 18:25 7m 10s
[JAX] Remove the redundant pjit BUILD target. CI #83505: Pull request #29126 synchronize by copybara-service bot
June 3, 2025 18:22 7m 2s test_765250212
test_765250212
June 3, 2025 18:22 7m 2s
[JAX] Remove the redundant pjit BUILD target. CI #83504: Commit cda50f5 pushed by copybara-service bot
June 3, 2025 18:22 2m 38s main
main
June 3, 2025 18:22 2m 38s
Add hermetic nvshmem implementation. CI #83503: Pull request #28892 synchronize by copybara-service bot
June 3, 2025 18:12 12m 9s test_753675848
test_753675848
June 3, 2025 18:12 12m 9s
[Mosaic GPU] Add BUILD rules for blackwell matmul kernel CI #83502: Pull request #29207 synchronize by copybara-service bot
June 3, 2025 18:08 12m 53s test_766711665
test_766711665
June 3, 2025 18:08 12m 53s
[Mosaic GPU] Add BUILD rules for blackwell matmul kernel CI #83501: Pull request #29207 synchronize by copybara-service bot
June 3, 2025 18:05 2m 27s test_766711665
test_766711665
June 3, 2025 18:05 2m 27s
[jaxlib] Add PyClient::Compile method that returns an unloaded PyExecutable. CI #83500: Pull request #29104 synchronize by copybara-service bot
June 3, 2025 18:01 12m 35s test_761950838
test_761950838
June 3, 2025 18:01 12m 35s
[Mosaic GPU] Add lowering for 2xf32 -> 2xf8e4m3fn conversions. CI #83499: Pull request #29198 synchronize by copybara-service bot
June 3, 2025 17:57 12m 14s test_766557719
test_766557719
June 3, 2025 17:57 12m 14s
[Mosaic GPU] Add lowering for 2xf32 -> 2xf8e4m3fn conversions. CI #83498: Commit e24f780 pushed by copybara-service bot
June 3, 2025 17:57 10m 0s main
main
June 3, 2025 17:57 10m 0s
[Pallas/Mosaic GPU] Expose the new TCGEN05_ROW layout. CI #83497: Pull request #29210 synchronize by copybara-service bot
June 3, 2025 17:56 10m 7s test_766735877
test_766735877
June 3, 2025 17:56 10m 7s
[pallas] In TPU interpret mode, run kernels in parallel over Megacore cores. CI #83496: Pull request #29187 synchronize by copybara-service bot
June 3, 2025 17:56 10m 2s test_765354228
test_765354228
June 3, 2025 17:56 10m 2s
[Pallas/Mosaic GPU] Expose the new TCGEN05_ROW layout. CI #83495: Pull request #29210 opened by copybara-service bot
June 3, 2025 17:55 1m 24s test_766735877
test_766735877
June 3, 2025 17:55 1m 24s
grad + unreduced CI #83494: Pull request #29209 opened by copybara-service bot
June 3, 2025 17:47 6m 20s test_756077108
test_756077108
June 3, 2025 17:47 6m 20s
Merge pull request #29186 from jakevdp:jax-numpy-imports CI #83493: Commit 0b89b23 pushed by copybara-service bot
June 3, 2025 17:45 10m 7s main
main
June 3, 2025 17:45 10m 7s
[Mosaic GPU] Add lowering for 2xf32 -> 2xf8e4m3fn conversions. CI #83492: Pull request #29198 synchronize by copybara-service bot
June 3, 2025 17:41 10m 7s test_766557719
test_766557719
June 3, 2025 17:41 10m 7s
fix missing input checks in jax.nn.functions CI #83491: Pull request #29206 synchronize by DanisNone
June 3, 2025 17:32 Action required DanisNone:main
DanisNone:main
June 3, 2025 17:32 Action required
[jaxlib] Add PyClient::Compile method that returns an unloaded PyExecutable. CI #83490: Pull request #29104 synchronize by copybara-service bot
June 3, 2025 17:25 9m 55s test_761950838
test_761950838
June 3, 2025 17:25 9m 55s
fix missing input checks in jax.nn.functions CI #83489: Pull request #29206 synchronize by DanisNone
June 3, 2025 17:22 Action required DanisNone:main
DanisNone:main
June 3, 2025 17:22 Action required
fix missing input checks in jax.nn.functions CI #83488: Pull request #29206 synchronize by DanisNone
June 3, 2025 17:18 Action required DanisNone:main
DanisNone:main
June 3, 2025 17:18 Action required
fix missing input checks in jax.nn.functions CI #83487: Pull request #29206 synchronize by DanisNone
June 3, 2025 17:17 Action required DanisNone:main
DanisNone:main
June 3, 2025 17:17 Action required