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

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Update XLA dependency to use revision JAX Array API #2530: Commit 33a7385 pushed by copybara-service bot
October 20, 2024 20:03 1m 1s main
main
October 20, 2024 20:03 1m 1s
Add an example for executing serialized function in cpp JAX Array API #2529: Pull request #24413 opened by fenghuizhang
October 20, 2024 17:38 Action required fenghuizhang:jax_cpp
fenghuizhang:jax_cpp
October 20, 2024 17:38 Action required
Better docs for jnp.fromfunction JAX Array API #2528: Pull request #24412 opened by jakevdp
October 20, 2024 16:21 54s jakevdp:fromfunction-doc
jakevdp:fromfunction-doc
October 20, 2024 16:21 54s
Remove mesh_utils.create_device_mesh from docs JAX Array API #2527: Pull request #24409 synchronize by copybara-service bot
October 19, 2024 22:48 58s test_687679312
test_687679312
October 19, 2024 22:48 58s
Remove mesh_utils.create_device_mesh from docs JAX Array API #2526: Commit ca2d158 pushed by copybara-service bot
October 19, 2024 22:48 53s main
main
October 19, 2024 22:48 53s
Remove mesh_utils.create_device_mesh from docs JAX Array API #2525: Pull request #24409 synchronize by copybara-service bot
October 19, 2024 22:36 55s test_687679312
test_687679312
October 19, 2024 22:36 55s
Remove mesh_utils.create_device_mesh from docs JAX Array API #2524: Pull request #24409 synchronize by copybara-service bot
October 19, 2024 22:29 53s test_687679312
test_687679312
October 19, 2024 22:29 53s
Remove mesh_utils.create_device_mesh from docs JAX Array API #2523: Pull request #24409 synchronize by copybara-service bot
October 19, 2024 22:12 54s test_687679312
test_687679312
October 19, 2024 22:12 54s
Remove mesh_utils.create_device_mesh from docs JAX Array API #2522: Pull request #24409 opened by copybara-service bot
October 19, 2024 22:06 55s test_687679312
test_687679312
October 19, 2024 22:06 55s
Update XLA dependency to use revision JAX Array API #2521: Commit 77fb1ee pushed by copybara-service bot
October 19, 2024 20:38 52s main
main
October 19, 2024 20:38 52s
Expand the ffi_call API to support more use cases JAX Array API #2520: Pull request #24370 synchronize by dfm
October 19, 2024 20:20 55s dfm:ffi-call-to-callable
dfm:ffi-call-to-callable
October 19, 2024 20:20 55s
Clean up conditional lowering of tan after JAX v0.4.34 release. JAX Array API #2519: Pull request #24405 opened by dfm
October 19, 2024 20:14 1m 0s dfm:cleanup-tan-lowering
dfm:cleanup-tan-lowering
October 19, 2024 20:14 1m 0s
Expand the ffi_call API to support more use cases JAX Array API #2518: Pull request #24370 synchronize by dfm
October 19, 2024 19:59 52s dfm:ffi-call-to-callable
dfm:ffi-call-to-callable
October 19, 2024 19:59 52s
Expand the ffi_call API to support more use cases JAX Array API #2517: Pull request #24370 synchronize by dfm
October 19, 2024 19:48 53s dfm:ffi-call-to-callable
dfm:ffi-call-to-callable
October 19, 2024 19:48 53s
Adds arith.select to the op patters in order to canonicalize non 32 bit selects. JAX Array API #2516: Pull request #24371 synchronize by copybara-service bot
October 19, 2024 16:09 52s test_683793926
test_683793926
October 19, 2024 16:09 52s
Adds arith.select to the op patters in order to canonicalize non 32 b… JAX Array API #2515: Commit 48bddc6 pushed by copybara-service bot
October 19, 2024 16:09 58s main
main
October 19, 2024 16:09 58s
Adds arith.select to the op patters in order to canonicalize non 32 bit selects. JAX Array API #2514: Pull request #24371 synchronize by copybara-service bot
October 19, 2024 15:30 58s test_683793926
test_683793926
October 19, 2024 15:30 58s
[jax2tf] Disable jax2tf with non-native serialization. JAX Array API #2513: Pull request #24183 synchronize by copybara-service bot
October 19, 2024 15:06 1m 1s test_683472001
test_683472001
October 19, 2024 15:06 1m 1s
Better documentation for jnp.load JAX Array API #2512: Pull request #24403 opened by jakevdp
October 19, 2024 13:20 55s jakevdp:load-doc
jakevdp:load-doc
October 19, 2024 13:20 55s
Add an extension mechanism to run_state that allows: JAX Array API #2511: Pull request #24386 synchronize by copybara-service bot
October 19, 2024 12:10 52s test_687287590
test_687287590
October 19, 2024 12:10 52s
Add an extension mechanism to run_state that allows: JAX Array API #2510: Pull request #24386 synchronize by copybara-service bot
October 19, 2024 11:59 1m 4s test_687287590
test_687287590
October 19, 2024 11:59 1m 4s
Add an extension mechanism to run_state that allows: JAX Array API #2509: Pull request #24386 synchronize by copybara-service bot
October 19, 2024 11:58 51s test_687287590
test_687287590
October 19, 2024 11:58 51s
Automated Code Change JAX Array API #2508: Pull request #24401 opened by copybara-service bot
October 19, 2024 08:52 55s test_685953062
test_685953062
October 19, 2024 08:52 55s
[export] Add support for serialization for some custom PyTree nodes JAX Array API #2507: Pull request #24342 synchronize by gnecula
October 19, 2024 08:46 55s gnecula:export_custom_types
gnecula:export_custom_types
October 19, 2024 08:46 55s
[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