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

[sharding_in_types] Add lax.transpose sharding propagation rule JAX Array API #2406: Commit 3e634d9 pushed by copybara-service bot
October 18, 2024 00:08 1m 4s main
main
October 18, 2024 00:08 1m 4s
Allow simple host call that uses host tensor as parameter/result in JAX Array API #2405: Pull request #24057 synchronize by copybara-service bot
October 18, 2024 00:05 1m 0s test_680683100
test_680683100
October 18, 2024 00:05 1m 0s
[sharding_in_types] Add lax.transpose sharding propagation rule JAX Array API #2404: Pull request #24377 synchronize by copybara-service bot
October 17, 2024 23:54 1m 0s test_687057762
test_687057762
October 17, 2024 23:54 1m 0s
[sharding_in_types] Support jnp.array with sharding_in_types. When the input array has a sharding, propagate it through without dropping the sharding. JAX Array API #2403: Pull request #24376 synchronize by copybara-service bot
October 17, 2024 23:51 53s test_687052240
test_687052240
October 17, 2024 23:51 53s
[sharding_in_types] Support jnp.array with sharding_in_types. When th… JAX Array API #2402: Commit 57a95a7 pushed by copybara-service bot
October 17, 2024 23:51 55s main
main
October 17, 2024 23:51 55s
[sharding_in_types] Support jnp.array with sharding_in_types. When the input array has a sharding, propagate it through without dropping the sharding. JAX Array API #2401: Pull request #24376 synchronize by copybara-service bot
October 17, 2024 23:40 57s test_687052240
test_687052240
October 17, 2024 23:40 57s
[sharding_in_types] Support jnp.array with sharding_in_types. When the input array has a sharding, propagate it through without dropping the sharding. JAX Array API #2400: Pull request #24376 synchronize by copybara-service bot
October 17, 2024 23:05 1m 4s test_687052240
test_687052240
October 17, 2024 23:05 1m 4s
[sharding_in_types] Add reduce max, integer_pow and standard_unop sharding rules JAX Array API #2399: Pull request #24375 synchronize by copybara-service bot
October 17, 2024 22:55 56s test_687034219
test_687034219
October 17, 2024 22:55 56s
[sharding_in_types] Add reduce max, integer_pow and standard_unop sha… JAX Array API #2398: Commit 5df4878 pushed by copybara-service bot
October 17, 2024 22:55 1m 0s main
main
October 17, 2024 22:55 1m 0s
[sharding_in_types] Add reduce max, integer_pow and standard_unop sharding rules JAX Array API #2397: Pull request #24375 synchronize by copybara-service bot
October 17, 2024 22:40 1m 11s test_687034219
test_687034219
October 17, 2024 22:40 1m 11s
[sharding_in_types] Add lax.transpose sharding propagation rule JAX Array API #2396: Pull request #24377 opened by copybara-service bot
October 17, 2024 22:40 1m 1s test_687057762
test_687057762
October 17, 2024 22:40 1m 1s
[sharding_in_types] Support jnp.array with sharding_in_types. When the input array has a sharding, propagate it through without dropping the sharding. JAX Array API #2395: Pull request #24376 opened by copybara-service bot
October 17, 2024 22:40 57s test_687052240
test_687052240
October 17, 2024 22:40 57s
Move converter related tflite functions to tensorflow/lite repo JAX Array API #2394: Pull request #24346 synchronize by copybara-service bot
October 17, 2024 22:08 59s test_686164263
test_686164263
October 17, 2024 22:08 59s
Move out mnist py/Jax tensorflow lite tests to tensorflow lite repo JAX Array API #2393: Pull request #24298 synchronize by copybara-service bot
October 17, 2024 22:07 1m 0s test_685822113
test_685822113
October 17, 2024 22:07 1m 0s
[sharding_in_types] Add broadcast_in_dim rule. JAX Array API #2392: Pull request #24374 synchronize by copybara-service bot
October 17, 2024 21:55 57s test_687014486
test_687014486
October 17, 2024 21:55 57s
[sharding_in_types] Add broadcast_in_dim rule. JAX Array API #2391: Commit e92e119 pushed by copybara-service bot
October 17, 2024 21:55 53s main
main
October 17, 2024 21:55 53s
[sharding_in_types] Add broadcast_in_dim rule. JAX Array API #2390: Pull request #24374 synchronize by copybara-service bot
October 17, 2024 21:43 59s test_687014486
test_687014486
October 17, 2024 21:43 59s
Update XLA dependency to use revision JAX Array API #2389: Commit 93389ab pushed by copybara-service bot
October 17, 2024 21:29 1m 0s main
main
October 17, 2024 21:29 1m 0s
[sharding_in_types] Add reduce max, integer_pow and standard_unop sharding rules JAX Array API #2388: Pull request #24375 opened by copybara-service bot
October 17, 2024 21:26 1m 7s test_687034219
test_687034219
October 17, 2024 21:26 1m 7s
Activate the FFI implementation of SVD on GPU. JAX Array API #2387: Pull request #24211 synchronize by copybara-service bot
October 17, 2024 21:14 1m 0s test_679137575
test_679137575
October 17, 2024 21:14 1m 0s
Test hero styling updates JAX Array API #2386: Pull request #24293 synchronize by barnesjoseph
October 17, 2024 21:03 Action required barnesjoseph:update-lp-styling
barnesjoseph:update-lp-styling
October 17, 2024 21:03 Action required
[sharding_in_types] Add broadcast_in_dim rule. JAX Array API #2385: Pull request #24374 synchronize by copybara-service bot
October 17, 2024 21:01 59s test_687014486
test_687014486
October 17, 2024 21:01 59s
Test hero styling updates JAX Array API #2384: Pull request #24293 synchronize by barnesjoseph
October 17, 2024 21:01 Action required barnesjoseph:update-lp-styling
barnesjoseph:update-lp-styling
October 17, 2024 21:01 Action required
Expand the ffi_call API to support more use cases JAX Array API #2383: Pull request #24370 synchronize by dfm
October 17, 2024 20:58 1m 1s dfm:ffi-call-to-callable
dfm:ffi-call-to-callable
October 17, 2024 20:58 1m 1s
Merge pull request #24345 from phu0ngng:cuda_custom_call JAX Array API #2382: Commit 919f7c8 pushed by copybara-service bot
October 17, 2024 20:57 55s main
main
October 17, 2024 20:57 55s