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

[export] Add support for serialization for some custom PyTree nodes JAX Array API #2506: Pull request #24342 synchronize by gnecula
October 19, 2024 04:53 52s gnecula:export_custom_types
gnecula:export_custom_types
October 19, 2024 04:53 52s
Make jnp.subtract a ufunc JAX Array API #2505: Pull request #24400 opened by jakevdp
October 19, 2024 03:34 58s jakevdp:subtract-ufunc
jakevdp:subtract-ufunc
October 19, 2024 03:34 58s
[Pallas TPU] Fix lowering for jnp.remainder JAX Array API #2504: Pull request #24356 synchronize by copybara-service bot
October 19, 2024 02:40 1m 0s test_686535642
test_686535642
October 19, 2024 02:40 1m 0s
resolve extended dtypes in a jaxpr pass JAX Array API #2503: Pull request #24302 synchronize by justinjfu
October 19, 2024 00:18 53s froystig:jax2jax
froystig:jax2jax
October 19, 2024 00:18 53s
resolve extended dtypes in a jaxpr pass JAX Array API #2502: Pull request #24302 synchronize by justinjfu
October 19, 2024 00:13 53s froystig:jax2jax
froystig:jax2jax
October 19, 2024 00:13 53s
[Mosaic TPU] Support dynamic DMA and ref slice on the 2nd minor when memref is untiled JAX Array API #2501: Pull request #24354 synchronize by copybara-service bot
October 18, 2024 23:53 52s test_684117211
test_684117211
October 18, 2024 23:53 52s
[Pallas TPU] Fix lowering for jnp.remainder JAX Array API #2500: Pull request #24356 synchronize by copybara-service bot
October 18, 2024 23:37 57s test_686535642
test_686535642
October 18, 2024 23:37 57s
[Pallas TPU] Use new MLIR op names JAX Array API #2499: Pull request #24388 synchronize by copybara-service bot
October 18, 2024 23:14 57s test_687278048
test_687278048
October 18, 2024 23:14 57s
[Pallas TPU] Use new MLIR op names JAX Array API #2498: Commit 884f1dc pushed by copybara-service bot
October 18, 2024 23:14 57s main
main
October 18, 2024 23:14 57s
[Pallas TPU] Use new MLIR op names JAX Array API #2497: Pull request #24388 synchronize by copybara-service bot
October 18, 2024 22:50 53s test_687278048
test_687278048
October 18, 2024 22:50 53s
resolve extended dtypes in a jaxpr pass JAX Array API #2496: Pull request #24302 synchronize by justinjfu
October 18, 2024 22:45 55s froystig:jax2jax
froystig:jax2jax
October 18, 2024 22:45 55s
resolve extended dtypes in a jaxpr pass JAX Array API #2495: Pull request #24302 synchronize by justinjfu
October 18, 2024 22:29 54s froystig:jax2jax
froystig:jax2jax
October 18, 2024 22:29 54s
resolve extended dtypes in a jaxpr pass JAX Array API #2494: Pull request #24302 synchronize by justinjfu
October 18, 2024 22:26 54s froystig:jax2jax
froystig:jax2jax
October 18, 2024 22:26 54s
resolve extended dtypes in a jaxpr pass JAX Array API #2493: Pull request #24302 synchronize by justinjfu
October 18, 2024 22:24 54s froystig:jax2jax
froystig:jax2jax
October 18, 2024 22:24 54s
resolve extended dtypes in a jaxpr pass JAX Array API #2492: Pull request #24302 synchronize by justinjfu
October 18, 2024 22:19 53s froystig:jax2jax
froystig:jax2jax
October 18, 2024 22:19 53s
resolve extended dtypes in a jaxpr pass JAX Array API #2491: Pull request #24302 synchronize by justinjfu
October 18, 2024 22:17 59s froystig:jax2jax
froystig:jax2jax
October 18, 2024 22:17 59s
resolve extended dtypes in a jaxpr pass JAX Array API #2490: Pull request #24302 synchronize by justinjfu
October 18, 2024 22:09 58s froystig:jax2jax
froystig:jax2jax
October 18, 2024 22:09 58s
Update XLA dependency to use revision JAX Array API #2489: Commit 2242651 pushed by copybara-service bot
October 18, 2024 21:35 54s main
main
October 18, 2024 21:35 54s
Avoid querying metadata query to check if it's GCE if TPU_SKIP_MDS_QUERY is set. JAX Array API #2488: Pull request #24394 synchronize by copybara-service bot
October 18, 2024 20:13 57s test_687399234
test_687399234
October 18, 2024 20:13 57s
Avoid querying metadata query to check if it's GCE if TPU_SKIP_MDS_QUERY is set. JAX Array API #2487: Pull request #24394 synchronize by copybara-service bot
October 18, 2024 20:10 55s test_687399234
test_687399234
October 18, 2024 20:10 55s
Avoid querying metadata query to check if it's GCE if TPU_SKIP_MDS_QUERY is set. JAX Array API #2486: Pull request #24394 synchronize by copybara-service bot
October 18, 2024 20:08 54s test_687399234
test_687399234
October 18, 2024 20:08 54s
Avoid querying metadata query to check if it's GCE if TPU_SKIP_MDS_QUERY is set. JAX Array API #2485: Pull request #24394 opened by copybara-service bot
October 18, 2024 20:07 59s test_687399234
test_687399234
October 18, 2024 20:07 59s
Fix env check order by splitting TPU envs into separate files. JAX Array API #2484: Pull request #24393 synchronize by copybara-service bot
October 18, 2024 19:59 59s test_687394503
test_687394503
October 18, 2024 19:59 59s
Fix env check order by splitting TPU envs into separate files. JAX Array API #2483: Pull request #24393 synchronize by copybara-service bot
October 18, 2024 19:55 39s test_687394503
test_687394503
October 18, 2024 19:55 39s
Fix env check order by splitting TPU envs into separate files. JAX Array API #2482: Pull request #24393 opened by copybara-service bot
October 18, 2024 19:52 36s test_687394503
test_687394503
October 18, 2024 19:52 36s