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
39,643 workflow runs
39,643 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Make c-api topology and PjRtClient versions produce identical platform_verison JAX Array API #43203: Pull request #33983 synchronize by copybara-service bot
1m 53s test_845452856
test_845452856
1m 53s
Add to_cotangent_aval to HiType and use it by default in bwd pass JAX Array API #43202: Pull request #33985 synchronize by copybara-service bot
4m 42s test_845470915
test_845470915
4m 42s
Add to_cotangent_aval to HiType and use it by default in bwd pass JAX Array API #43201: Commit 9a391d7 pushed by copybara-service bot
2m 4s main
main
2m 4s
[XLA:TPU] Abort early if the peak HLO memory consumption is greater than HBM JAX Array API #43200: Pull request #33986 opened by copybara-service bot
2m 16s test_843851238
test_843851238
2m 16s
Add to_cotangent_aval to HiType and use it by default in bwd pass JAX Array API #43199: Pull request #33985 synchronize by copybara-service bot
1m 58s test_845470915
test_845470915
1m 58s
Add to_cotangent_aval to HiType and use it by default in bwd pass JAX Array API #43198: Pull request #33985 opened by copybara-service bot
1m 59s test_845470915
test_845470915
1m 59s
Merge pull request #33552 from Ma-gi-cian:poisson-entropy JAX Array API #43197: Commit 8d4e9c1 pushed by copybara-service bot
2m 22s main
main
2m 22s
jax.numpy: return tuples to match NumPy 2.0 JAX Array API #43196: Pull request #33984 opened by jakevdp
1m 57s jakevdp:return-tuple
jakevdp:return-tuple
1m 57s
Make c-api topology and PjRtClient versions produce identical platform_verison JAX Array API #43195: Pull request #33983 opened by copybara-service bot
2m 18s test_845452856
test_845452856
2m 18s
[Pallas] Allow no mesh context if just signaling on core axis JAX Array API #43194: Pull request #33980 synchronize by copybara-service bot
4m 42s test_845362276
test_845362276
4m 42s
[Pallas] Allow no mesh context if just signaling on core axis JAX Array API #43193: Commit ad2b914 pushed by copybara-service bot
4m 44s main
main
4m 44s
Implement higher order interpolations in jax.scipy.ndimage JAX Array API #43192: Pull request #33970 synchronize by hexane360
Action required hexane360:ndimage-higher-order
hexane360:ndimage-higher-order
Action required
[Pallas MGPU] Simplify how we keep track of the current output slices. Keeping track of the full store slices is unnecessary because the slice size doesn’t change, we really only care about the the start of the slices. JAX Array API #43191: Pull request #33982 opened by copybara-service bot
1m 47s test_845422049
test_845422049
1m 47s
[Pallas MGPU] Adding support for squeezed block dims in the pipeline BlockSpecs. They can be identified with a None or pl.Squeezed. JAX Array API #43190: Pull request #33713 synchronize by copybara-service bot
2m 22s test_839975575
test_839975575
2m 22s
Skip test_scalar_debug_check in tpu_sparsecore_pallas_debug_check_test. JAX Array API #43189: Pull request #33981 opened by copybara-service bot
2m 18s test_845416829
test_845416829
2m 18s
ASAN tests JAX Array API #43188: Pull request #33774 synchronize by copybara-service bot
2m 21s test_840749908
test_840749908
2m 21s
ASAN tests JAX Array API #43187: Pull request #33774 synchronize by copybara-service bot
2m 26s test_840749908
test_840749908
2m 26s
Refactor indexing code JAX Array API #43186: Pull request #33752 synchronize by jakevdp
1m 51s jakevdp:indexing-refactor
jakevdp:indexing-refactor
1m 51s
Bisect failing callback test JAX Array API #43185: Pull request #33979 synchronize by copybara-service bot
2m 45s test_845350990
test_845350990
2m 45s
[Pallas] Allow no mesh context if just signaling on core axis JAX Array API #43184: Pull request #33980 synchronize by copybara-service bot
2m 22s test_845362276
test_845362276
2m 22s
Remove dynamic grid bounds restriction in Pallas Mosaic with memory spaces. JAX Array API #43183: Pull request #33962 synchronize by copybara-service bot
4m 5s test_844836772
test_844836772
4m 5s
Remove dynamic grid bounds restriction in Pallas Mosaic with memory s… JAX Array API #43182: Commit ba024e3 pushed by copybara-service bot
1m 57s main
main
1m 57s
Implementing PReLu JAX Array API #43181: Pull request #29147 synchronize by MythicArrow
Action required MythicArrow:patch-3
MythicArrow:patch-3
Action required
[Pallas] Allow no mesh context if just signaling on core axis JAX Array API #43180: Pull request #33980 synchronize by copybara-service bot
2m 0s test_845362276
test_845362276
2m 0s
[Pallas] Allow no mesh context if just signaling on core axis JAX Array API #43179: Pull request #33980 synchronize by copybara-service bot
1m 53s test_845362276
test_845362276
1m 53s