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
10,963 workflow runs
10,963 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[JAX] Implement an initial object API for colocated Python JAX Array API #11432: Pull request #26626 synchronize by copybara-service bot
February 21, 2025 20:45 2m 46s test_720733692
test_720733692
February 21, 2025 20:45 2m 46s
Implement Jax CPU/GPU callbacks with XLA's FFI. JAX Array API #11431: Pull request #26167 synchronize by copybara-service bot
February 21, 2025 20:00 2m 52s test_720696248
test_720696248
February 21, 2025 20:00 2m 52s
Reverts 655267609b5589fda8358ae7aaf2eb832036407a JAX Array API #11430: Commit 6c83d43 pushed by copybara-service bot
February 21, 2025 19:59 2m 56s main
main
February 21, 2025 19:59 2m 56s
Enable :debug_info_test/test_vjp_of_jit. JAX Array API #11429: Pull request #26671 synchronize by copybara-service bot
February 21, 2025 19:54 2m 59s test_729588219
test_729588219
February 21, 2025 19:54 2m 59s
Enable :debug_info_test/test_vjp_of_jit. JAX Array API #11428: Pull request #26671 synchronize by copybara-service bot
February 21, 2025 19:43 2m 47s test_729588219
test_729588219
February 21, 2025 19:43 2m 47s
Enable :debug_info_test/test_vjp_of_jit. JAX Array API #11427: Pull request #26671 synchronize by copybara-service bot
February 21, 2025 19:37 2m 52s test_729588219
test_729588219
February 21, 2025 19:37 2m 52s
Enable :debug_info_test/test_vjp_of_jit. JAX Array API #11426: Pull request #26671 synchronize by copybara-service bot
February 21, 2025 19:31 2m 50s test_729588219
test_729588219
February 21, 2025 19:31 2m 50s
Proof-of-concept: Use XLA's intra_op_threadpool to batch parallelize JAX's LAPACK calls on CPU. JAX Array API #11425: Pull request #26280 synchronize by copybara-service bot
February 21, 2025 19:19 2m 48s test_722714201
test_722714201
February 21, 2025 19:19 2m 48s
Enable :debug_info_test/test_vjp_of_jit. JAX Array API #11424: Pull request #26671 synchronize by copybara-service bot
February 21, 2025 19:19 3m 6s test_729588219
test_729588219
February 21, 2025 19:19 3m 6s
Enable :debug_info_test/test_vjp_of_jit. JAX Array API #11423: Pull request #26671 synchronize by copybara-service bot
February 21, 2025 19:17 2m 23s test_729588219
test_729588219
February 21, 2025 19:17 2m 23s
Enable :debug_info_test/test_vjp_of_jit. JAX Array API #11422: Pull request #26671 opened by copybara-service bot
February 21, 2025 19:03 2m 59s test_729588219
test_729588219
February 21, 2025 19:03 2m 59s
Remove CUDA dependencies from jaxlib OSS wheel. JAX Array API #11421: Pull request #25485 synchronize by copybara-service bot
February 21, 2025 19:01 2m 53s test_706016685
test_706016685
February 21, 2025 19:01 2m 53s
Reverts 655267609b5589fda8358ae7aaf2eb832036407a JAX Array API #11420: Pull request #26663 synchronize by copybara-service bot
February 21, 2025 18:56 2m 48s test_729520942
test_729520942
February 21, 2025 18:56 2m 48s
Redefine is_fully_addressable in shardings to support zero local devices for McJAX. JAX Array API #11419: Pull request #26670 synchronize by copybara-service bot
February 21, 2025 18:38 2m 52s test_729574794
test_729574794
February 21, 2025 18:38 2m 52s
Make mesh and *_spec parameters optional. JAX Array API #11418: Pull request #26669 synchronize by copybara-service bot
February 21, 2025 18:35 2m 52s test_729551793
test_729551793
February 21, 2025 18:35 2m 52s
Redefine is_fully_addressable in shardings to support zero local devices for McJAX. JAX Array API #11417: Pull request #26670 opened by copybara-service bot
February 21, 2025 18:32 2m 58s test_729574794
test_729574794
February 21, 2025 18:32 2m 58s
[JAX] Implement an initial object API for colocated Python JAX Array API #11416: Pull request #26626 synchronize by copybara-service bot
February 21, 2025 18:22 2m 48s test_720733692
test_720733692
February 21, 2025 18:22 2m 48s
Update several remaining lax.linalg primitives to use registration he… JAX Array API #11415: Commit c4418c1 pushed by copybara-service bot
February 21, 2025 18:18 3m 15s main
main
February 21, 2025 18:18 3m 15s
Use new ML Build CUDA images JAX Array API #11414: Commit 5089fb0 pushed by copybara-service bot
February 21, 2025 18:16 2m 32s main
main
February 21, 2025 18:16 2m 32s
Update Windows docker image to pin it to a sha JAX Array API #11413: Commit a073639 pushed by copybara-service bot
February 21, 2025 18:14 2m 42s main
main
February 21, 2025 18:14 2m 42s
Implement Jax CPU/GPU callbacks with XLA's FFI. JAX Array API #11412: Pull request #26167 synchronize by copybara-service bot
February 21, 2025 18:06 2m 46s test_720696248
test_720696248
February 21, 2025 18:06 2m 46s
Make mesh and *_spec parameters optional. JAX Array API #11411: Pull request #26669 synchronize by copybara-service bot
February 21, 2025 18:04 2m 50s test_729551793
test_729551793
February 21, 2025 18:04 2m 50s
Update several remaining lax.linalg primitives to use registration helper functions. JAX Array API #11410: Pull request #26611 synchronize by copybara-service bot
February 21, 2025 17:56 2m 58s test_728268827
test_728268827
February 21, 2025 17:56 2m 58s
Use new ML Build CUDA images JAX Array API #11409: Pull request #26644 synchronize by copybara-service bot
February 21, 2025 17:52 2m 49s test_729207975
test_729207975
February 21, 2025 17:52 2m 49s
Update Windows docker image to pin it to a sha JAX Array API #11408: Pull request #26635 synchronize by copybara-service bot
February 21, 2025 17:52 2m 50s test_729125349
test_729125349
February 21, 2025 17:52 2m 50s