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

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[ROCm] Update Skip Reason Output for run_on_devices decorator JAX Array API #46261: Pull request #34675 opened by gulsumgudukbay
Action required ROCm:skip-change-rocm
ROCm:skip-change-rocm
Action required
[pmap] Speed up the new jax.pmap implementation. JAX Array API #46260: Pull request #34457 synchronize by copybara-service bot
4m 21s test_857829844
test_857829844
4m 21s
[pmap] Speed up the new jax.pmap implementation. JAX Array API #46259: Commit cf76313 pushed by copybara-service bot
2m 18s main
main
2m 18s
fix(scipy): consistently return float and handle nan propagation in rankdata JAX Array API #46258: Pull request #34628 synchronize by abdulwahabahmedkhanyusufzai
Action required abdulwahabahmedkhanyusufzai:fix-rankdata-nan-propagation
abdulwahabahmedkhanyusufzai:fix-rankdata-nan-propagation
Action required
[Mosaic] Update logistic lowering to correctly handle bf16 JAX Array API #46257: Pull request #34674 opened by catswe
Action required catswe:update_logistic_lowering
catswe:update_logistic_lowering
Action required
[pmap] Speed up the new jax.pmap implementation. JAX Array API #46256: Pull request #34457 synchronize by copybara-service bot
2m 21s test_857829844
test_857829844
2m 21s
[pmap] Optimize host-local <> global conversions. JAX Array API #46255: Pull request #34525 synchronize by copybara-service bot
2m 21s test_858926170
test_858926170
2m 21s
[pmap] Optimize host-local <> global conversions. JAX Array API #46254: Commit a2912b0 pushed by copybara-service bot
2m 17s main
main
2m 17s
Fix: Use ASAN feature instead of config setting JAX Array API #46253: Pull request #34647 synchronize by copybara-service bot
4m 5s test_861430825
test_861430825
4m 5s
Fix: Use ASAN feature instead of config setting JAX Array API #46252: Commit 678acf2 pushed by copybara-service bot
1m 53s main
main
1m 53s
[pmap] Optimize host-local <> global conversions. JAX Array API #46251: Pull request #34525 synchronize by copybara-service bot
2m 21s test_858926170
test_858926170
2m 21s
Fix: Use ASAN feature instead of config setting JAX Array API #46250: Pull request #34647 synchronize by copybara-service bot
2m 16s test_861430825
test_861430825
2m 16s
Add JAX version metadata to xspace JAX Array API #46249: Pull request #33740 synchronize by copybara-service bot
1m 56s test_840138208
test_840138208
1m 56s
[pmap] Push more into _cached_shard_map. JAX Array API #46248: Pull request #34524 synchronize by copybara-service bot
4m 20s test_858915173
test_858915173
4m 20s
[pmap] Push more into _cached_shard_map. JAX Array API #46247: Commit ee91164 pushed by copybara-service bot
2m 21s main
main
2m 21s
[Mosaic GPU] Fix handling of sliced/transposed types in transformed_smem_ref_type. JAX Array API #46246: Pull request #34673 opened by copybara-service bot
2m 0s test_861854584
test_861854584
2m 0s
[Mosaic TPU] Relax offset alignment requirement during mask generation for packed types in SingleRowVRegBounds. JAX Array API #46245: Pull request #34585 synchronize by copybara-service bot
2m 21s test_859825926
test_859825926
2m 21s
[Mosaic TPU] Relax offset alignment requirement during mask generation for packed types in SingleRowVRegBounds. JAX Array API #46244: Pull request #34585 synchronize by copybara-service bot
1m 54s test_859825926
test_859825926
1m 54s
Add JAX version metadata to xspace JAX Array API #46243: Pull request #33740 synchronize by copybara-service bot
2m 0s test_840138208
test_840138208
2m 0s
[Mosaic] Expose tpu::isGather JAX Array API #46242: Pull request #34116 synchronize by copybara-service bot
4m 28s test_848299928
test_848299928
4m 28s
[Mosaic] Expose tpu::isGather JAX Array API #46241: Commit 2dbb22d pushed by copybara-service bot
1m 49s main
main
1m 49s
[Pallas/interpreter] Support interpreting non-trivial GPU grids. JAX Array API #46240: Pull request #34590 synchronize by copybara-service bot
1m 55s test_858196665
test_858196665
1m 55s
Add JAX version metadata to xspace JAX Array API #46239: Pull request #33740 synchronize by copybara-service bot
2m 15s test_840138208
test_840138208
2m 15s
Add JAX version metadata to xspace JAX Array API #46238: Pull request #33740 synchronize by copybara-service bot
2m 7s test_840138208
test_840138208
2m 7s
Add env variables JAX_TRITON_PTXAS_DEVICE_DEBUG JAX_TRITON_PTXAS_EXTRA_FLAGS. JAX Array API #46237: Pull request #34539 synchronize by copybara-service bot
1m 56s test_858777974
test_858777974
1m 56s