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

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Error out if going from Manual -> Auto/Explicit AxisTypes in the auto_axes and explicit_axes API that do mesh_cast implicitly. JAX Array API #11183: Pull request #26606 synchronize by copybara-service bot
February 19, 2025 16:59 2m 55s test_728663881
test_728663881
February 19, 2025 16:59 2m 55s
roofline: Support computing flops for binary ops. JAX Array API #11182: Pull request #26589 synchronize by copybara-service bot
February 19, 2025 16:56 2m 53s test_728319438
test_728319438
February 19, 2025 16:56 2m 53s
[StableHLO] Only emit StableHLO from xla_computation_to_mlir_module JAX Array API #11181: Pull request #26585 synchronize by copybara-service bot
February 19, 2025 16:51 2m 51s test_727004976
test_727004976
February 19, 2025 16:51 2m 51s
[StableHLO] Only emit StableHLO from xla_computation_to_mlir_module JAX Array API #11180: Pull request #26585 synchronize by copybara-service bot
February 19, 2025 16:44 2m 58s test_727004976
test_727004976
February 19, 2025 16:44 2m 58s
Add dtype dispatching to FFI example. JAX Array API #11179: Pull request #26607 opened by dfm
February 19, 2025 16:39 2m 58s dfm:ffi-example-dtype-dispatch
dfm:ffi-example-dtype-dispatch
February 19, 2025 16:39 2m 58s
Create JAX wheel build target. JAX Array API #11178: Pull request #26378 synchronize by copybara-service bot
February 19, 2025 16:21 2m 52s test_724102315
test_724102315
February 19, 2025 16:21 2m 52s
Update internal unop primitive helper to pass kwargs to dtype rule. JAX Array API #11177: Pull request #26587 synchronize by copybara-service bot
February 19, 2025 16:03 2m 58s test_728296067
test_728296067
February 19, 2025 16:03 2m 58s
Upgrade to Bazel 7.x JAX Array API #11176: Pull request #26513 synchronize by copybara-service bot
February 19, 2025 16:02 2m 59s test_726036888
test_726036888
February 19, 2025 16:02 2m 59s
[better_errors] Simplify api_util.debug_info. JAX Array API #11175: Pull request #26603 reopened by gnecula
February 19, 2025 15:51 2m 58s gnecula:debug_info_opt1
gnecula:debug_info_opt1
February 19, 2025 15:51 2m 58s
Error out if going from Manual -> Auto/Explicit AxisTypes in the auto_axes and explicit_axes API that do mesh_cast implicitly. JAX Array API #11174: Pull request #26606 opened by copybara-service bot
February 19, 2025 15:37 2m 50s test_728663881
test_728663881
February 19, 2025 15:37 2m 50s
Use "common" instead of "build" for some flags in .bazelrc JAX Array API #11173: Pull request #26601 synchronize by copybara-service bot
February 19, 2025 15:09 2m 50s test_728592489
test_728592489
February 19, 2025 15:09 2m 50s
Use "common" instead of "build" for some flags in .bazelrc JAX Array API #11172: Commit 679f1a9 pushed by copybara-service bot
February 19, 2025 15:09 2m 46s main
main
February 19, 2025 15:09 2m 46s
Now that sharding_in_types config flag is True, remove the config and all the conditionals JAX Array API #11171: Pull request #26595 synchronize by copybara-service bot
February 19, 2025 14:53 2m 52s test_728413540
test_728413540
February 19, 2025 14:53 2m 52s
Now that sharding_in_types config flag is True, remove the config and… JAX Array API #11170: Commit a3edfb4 pushed by copybara-service bot
February 19, 2025 14:53 2m 55s main
main
February 19, 2025 14:53 2m 55s
Now that sharding_in_types config flag is True, remove the config and all the conditionals JAX Array API #11169: Pull request #26595 synchronize by copybara-service bot
February 19, 2025 14:37 2m 58s test_728413540
test_728413540
February 19, 2025 14:37 2m 58s
Now that sharding_in_types config flag is True, remove the config and all the conditionals JAX Array API #11168: Pull request #26595 synchronize by copybara-service bot
February 19, 2025 13:18 2m 53s test_728413540
test_728413540
February 19, 2025 13:18 2m 53s
[pallas:triton] Fix atomic min/max lowering for uint and float types JAX Array API #11167: Pull request #26604 opened by draganmladjenovic
February 19, 2025 12:55 Action required ROCm:min_max_atomics
ROCm:min_max_atomics
February 19, 2025 12:55 Action required
Use "common" instead of "build" for some flags in .bazelrc JAX Array API #11166: Pull request #26601 synchronize by copybara-service bot
February 19, 2025 12:50 2m 45s test_728592489
test_728592489
February 19, 2025 12:50 2m 45s
[better_errors] Simplify api_util.debug_info. JAX Array API #11165: Pull request #26603 synchronize by gnecula
February 19, 2025 12:47 2m 46s gnecula:debug_info_opt1
gnecula:debug_info_opt1
February 19, 2025 12:47 2m 46s
[better_errors] Simplify api_util.debug_info. JAX Array API #11164: Pull request #26603 opened by gnecula
February 19, 2025 12:44 3m 5s gnecula:debug_info_opt1
gnecula:debug_info_opt1
February 19, 2025 12:44 3m 5s
Use consistent dtype for forward and backwards in jax.nn.dot_product_… JAX Array API #11163: Commit d5e5b42 pushed by copybara-service bot
February 19, 2025 12:30 2m 52s main
main
February 19, 2025 12:30 2m 52s
Use consistent dtype for forward and backwards in jax.nn.dot_product_attention. JAX Array API #11162: Pull request #25433 synchronize by copybara-service bot
February 19, 2025 12:14 3m 0s test_705452797
test_705452797
February 19, 2025 12:14 3m 0s
Update XLA dependency to use revision JAX Array API #11161: Commit 4487229 pushed by copybara-service bot
February 19, 2025 12:05 2m 49s main
main
February 19, 2025 12:05 2m 49s
Use consistent dtype for forward and backwards in jax.nn.dot_product_attention. JAX Array API #11160: Pull request #25433 synchronize by copybara-service bot
February 19, 2025 11:52 2m 47s test_705452797
test_705452797
February 19, 2025 11:52 2m 47s
Add jax.experimental._mini_mpmd JAX Array API #11159: Pull request #26564 synchronize by gspschmid
February 19, 2025 11:40 Action required gspschmid:gschmid/mini_mpmd
gspschmid:gschmid/mini_mpmd
February 19, 2025 11:40 Action required