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
2,505 workflow runs
2,505 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Pallas TPU] Use new MLIR op names JAX Array API #2456: Pull request #24388 synchronize by copybara-service bot
October 18, 2024 15:31 1m 2s test_687278048
test_687278048
October 18, 2024 15:31 1m 2s
[Pallas TPU] Use new MLIR op names JAX Array API #2455: Pull request #24388 opened by copybara-service bot
October 18, 2024 15:24 57s test_687278048
test_687278048
October 18, 2024 15:24 57s
[Pallas:MGPU] Fix docstring for commit_shared JAX Array API #2454: Pull request #24384 synchronize by copybara-service bot
October 18, 2024 15:17 1m 7s test_687256987
test_687256987
October 18, 2024 15:17 1m 7s
[Pallas:MGPU] Fix docstring for commit_shared JAX Array API #2453: Commit e138e8e pushed by copybara-service bot
October 18, 2024 15:17 1m 6s main
main
October 18, 2024 15:17 1m 6s
[sharding_in_types] If out_aval.sharding is not None and the user specified out_sharding is None, concretize it with the device assignment available and add it to the final out_shardings that's used for lowering and compilation. JAX Array API #2452: Pull request #24387 opened by copybara-service bot
October 18, 2024 15:13 1m 11s test_687174276
test_687174276
October 18, 2024 15:13 1m 11s
[Pallas:MGPU] Fix docstring for commit_shared JAX Array API #2451: Pull request #24384 synchronize by copybara-service bot
October 18, 2024 15:02 1m 3s test_687256987
test_687256987
October 18, 2024 15:02 1m 3s
[pallas] run_scoped now supports partial discharge. JAX Array API #2450: Pull request #24344 synchronize by copybara-service bot
October 18, 2024 14:25 1m 3s test_686558966
test_686558966
October 18, 2024 14:25 1m 3s
aarch64: add aarch64 mkldnn+acl build config JAX Array API #2449: Pull request #23225 synchronize by snadampal
October 18, 2024 14:01 Action required snadampal:aarch64_jax
snadampal:aarch64_jax
October 18, 2024 14:01 Action required
Add an extension mechanism to run_state that allows: JAX Array API #2448: Pull request #24386 opened by copybara-service bot
October 18, 2024 13:56 51s test_687287590
test_687287590
October 18, 2024 13:56 51s
Add a dialect for Mosaic GPU. JAX Array API #2447: Pull request #24385 opened by copybara-service bot
October 18, 2024 13:50 1m 2s test_686535549
test_686535549
October 18, 2024 13:50 1m 2s
[Pallas TPU] Add lowerings for bf16 jnp.ceil and jnp.floor in TPU v6+ JAX Array API #2446: Pull request #24310 synchronize by copybara-service bot
October 18, 2024 13:45 58s test_686094131
test_686094131
October 18, 2024 13:45 58s
[Pallas TPU] Add lowerings for bf16 jnp.ceil and jnp.floor in TPU v6+ JAX Array API #2445: Pull request #24310 synchronize by copybara-service bot
October 18, 2024 13:04 54s test_686094131
test_686094131
October 18, 2024 13:04 54s
nit: Use frozen dataclasses rather than unsafe_hash. JAX Array API #2444: Pull request #24369 synchronize by copybara-service bot
October 18, 2024 12:36 1m 2s test_686926206
test_686926206
October 18, 2024 12:36 1m 2s
nit: Use frozen dataclasses rather than unsafe_hash. JAX Array API #2443: Commit 8615556 pushed by copybara-service bot
October 18, 2024 12:36 53s main
main
October 18, 2024 12:36 53s
[Pallas:MGPU] Force alignment of SMEM allocations to 1024 bytes JAX Array API #2442: Commit 4094564 pushed by copybara-service bot
October 18, 2024 12:21 1m 13s main
main
October 18, 2024 12:21 1m 13s
nit: Use frozen dataclasses rather than unsafe_hash. JAX Array API #2441: Pull request #24369 synchronize by copybara-service bot
October 18, 2024 12:13 54s test_686926206
test_686926206
October 18, 2024 12:13 54s
[Pallas:MGPU] Fix docstring for commit_shared JAX Array API #2440: Pull request #24384 opened by copybara-service bot
October 18, 2024 12:10 59s test_687256987
test_687256987
October 18, 2024 12:10 59s
[Pallas:MGPU] Allow indexing to appear anywhere in the list of transforms JAX Array API #2439: Pull request #24383 opened by copybara-service bot
October 18, 2024 12:09 52s test_687261076
test_687261076
October 18, 2024 12:09 52s
[Pallas:MGPU] Force alignment of SMEM allocations to 1024 bytes JAX Array API #2438: Pull request #24381 synchronize by copybara-service bot
October 18, 2024 12:09 1m 6s test_687243453
test_687243453
October 18, 2024 12:09 1m 6s
[Pallas:MGPU] Fix the implementation of WGMMA with transposed RHS JAX Array API #2437: Pull request #24382 opened by copybara-service bot
October 18, 2024 12:09 56s test_687261078
test_687261078
October 18, 2024 12:09 56s
[Pallas TPU] Add lowerings for bf16 jnp.ceil and jnp.floor in TPU v6+ JAX Array API #2436: Pull request #24310 synchronize by copybara-service bot
October 18, 2024 11:57 54s test_686094131
test_686094131
October 18, 2024 11:57 54s
[Pallas:MGPU] Add support for indexed refs to WGMMA JAX Array API #2435: Pull request #24363 synchronize by copybara-service bot
October 18, 2024 11:55 53s test_686902376
test_686902376
October 18, 2024 11:55 53s
[Pallas:MGPU] Add support for indexed refs to WGMMA JAX Array API #2434: Commit 0ee9531 pushed by copybara-service bot
October 18, 2024 11:55 53s main
main
October 18, 2024 11:55 53s
[Pallas:MGPU] Add support for indexed refs to WGMMA JAX Array API #2433: Pull request #24363 synchronize by copybara-service bot
October 18, 2024 11:44 55s test_686902376
test_686902376
October 18, 2024 11:44 55s
[Pallas TPU] Add lowerings for bf16 jnp.ceil and jnp.floor in TPU v6+ JAX Array API #2432: Pull request #24310 synchronize by copybara-service bot
October 18, 2024 11:27 55s test_686094131
test_686094131
October 18, 2024 11:27 55s