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
4,407 workflow runs
4,407 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Merge pull request #24899 from vfdev-5:patch-2 JAX Array API #4464: Commit 3380a21 pushed by copybara-service bot
November 14, 2024 17:43 54s main
main
November 14, 2024 17:43 54s
Implement lax.pad in Pallas. JAX Array API #4463: Pull request #24890 synchronize by copybara-service bot
November 14, 2024 17:29 58s test_696226549
test_696226549
November 14, 2024 17:29 58s
[Mosaic][TPU] Omit short circuiting of relayout (we should always relayout!) and implement product mismatch case for where we relayout from replicated to offset, and the number of vregs changes. JAX Array API #4462: Pull request #24887 synchronize by copybara-service bot
November 14, 2024 17:29 55s test_696226548
test_696226548
November 14, 2024 17:29 55s
Fix pgle test handling of directory removal. JAX Array API #4461: Pull request #24893 synchronize by copybara-service bot
November 14, 2024 17:23 1m 2s test_696369639
test_696369639
November 14, 2024 17:23 1m 2s
finalize deprecation of jax.clear_backends JAX Array API #4460: Pull request #24865 synchronize by jakevdp
November 14, 2024 17:22 1m 1s jakevdp:dep-clear-backends
jakevdp:dep-clear-backends
November 14, 2024 17:22 1m 1s
Fixed typo in asan.yaml JAX Array API #4459: Pull request #24899 opened by vfdev-5
November 14, 2024 17:00 1m 2s vfdev-5:patch-2
vfdev-5:patch-2
November 14, 2024 17:00 1m 2s
Added CI job with TSAN and free-threading JAX Array API #4458: Pull request #24898 synchronize by vfdev-5
November 14, 2024 16:58 58s vfdev-5:add-tsan-ft-ci-job
vfdev-5:add-tsan-ft-ci-job
November 14, 2024 16:58 58s
Added CI job with TSAN and free-threading JAX Array API #4457: Pull request #24898 synchronize by vfdev-5
November 14, 2024 16:57 56s vfdev-5:add-tsan-ft-ci-job
vfdev-5:add-tsan-ft-ci-job
November 14, 2024 16:57 56s
Merge pull request #23374 from jaro-sevcik:mock-topology-config JAX Array API #4456: Commit a1eb5ce pushed by copybara-service bot
November 14, 2024 16:55 1m 2s main
main
November 14, 2024 16:55 1m 2s
Added CI job with TSAN and free-threading JAX Array API #4455: Pull request #24898 opened by vfdev-5
November 14, 2024 16:39 52s vfdev-5:add-tsan-ft-ci-job
vfdev-5:add-tsan-ft-ci-job
November 14, 2024 16:39 52s
Don't use an out-of-line lowering for integer_pow for small powers. JAX Array API #4454: Pull request #24897 opened by hawkinsp
November 14, 2024 16:23 58s hawkinsp:ipow
hawkinsp:ipow
November 14, 2024 16:23 58s
Make the jaxpr for jnp.pad in "constant" mode more succinct. JAX Array API #4453: Pull request #24896 synchronize by hawkinsp
November 14, 2024 16:03 55s hawkinsp:pad
hawkinsp:pad
November 14, 2024 16:03 55s
Fix pgle test handling of directory removal. JAX Array API #4452: Pull request #24893 synchronize by copybara-service bot
November 14, 2024 15:49 57s test_696369639
test_696369639
November 14, 2024 15:49 57s
Make the jaxpr for jnp.pad in "constant" mode more succinct. JAX Array API #4451: Pull request #24896 synchronize by hawkinsp
November 14, 2024 15:37 1m 0s hawkinsp:pad
hawkinsp:pad
November 14, 2024 15:37 1m 0s
Make the jaxpr for jnp.pad in "constant" mode more succinct. JAX Array API #4450: Pull request #24896 opened by hawkinsp
November 14, 2024 15:05 1m 2s hawkinsp:pad
hawkinsp:pad
November 14, 2024 15:05 1m 2s
[Mosaic TPU] Support 1D concat: set implicit_dim to kSecondMinor to treat 1D (N,) as (1, N) and then tile it as (1, 128) JAX Array API #4449: Pull request #24781 synchronize by copybara-service bot
November 14, 2024 10:44 56s test_694460391
test_694460391
November 14, 2024 10:44 56s
[pallas:mosaic_gpu] Ported two pipelining optimizations to emit_pipeline JAX Array API #4448: Pull request #24590 synchronize by copybara-service bot
November 14, 2024 10:37 51s test_691125207
test_691125207
November 14, 2024 10:37 51s
[pallas:mosaic_gpu] Ported two pipelining optimizations to `emit_pipe… JAX Array API #4447: Commit aefe621 pushed by copybara-service bot
November 14, 2024 10:37 1m 2s main
main
November 14, 2024 10:37 1m 2s
[pallas:mosaic_gpu] Ported two pipelining optimizations to emit_pipeline JAX Array API #4446: Pull request #24590 synchronize by copybara-service bot
November 14, 2024 10:23 57s test_691125207
test_691125207
November 14, 2024 10:23 57s
Determine symmetric linear op fro CG from abstract output dtype JAX Array API #4445: Pull request #23486 synchronize by Joshuaalbert
November 14, 2024 10:20 Action required Joshuaalbert:main
Joshuaalbert:main
November 14, 2024 10:20 Action required
[pallas:mosaic_gpu] Ported two pipelining optimizations to emit_pipeline JAX Array API #4444: Pull request #24590 synchronize by copybara-service bot
November 14, 2024 10:13 1m 7s test_691125207
test_691125207
November 14, 2024 10:13 1m 7s
Implement lax.pad in Pallas. JAX Array API #4443: Pull request #24890 synchronize by copybara-service bot
November 14, 2024 08:11 1m 2s test_696226549
test_696226549
November 14, 2024 08:11 1m 2s
[Mosaic][TPU] Omit short circuiting of relayout (we should always relayout!) and implement product mismatch case for where we relayout from replicated to offset, and the number of vregs changes. JAX Array API #4442: Pull request #24887 synchronize by copybara-service bot
November 14, 2024 07:50 59s test_696226548
test_696226548
November 14, 2024 07:50 59s
[sharding_in_types] Support shard_map with sharding in types. Right now only full manual mode is supported. JAX Array API #4441: Pull request #24894 synchronize by copybara-service bot
November 14, 2024 06:41 1m 0s test_693048291
test_693048291
November 14, 2024 06:41 1m 0s
[sharding_in_types] Support shard_map with sharding in types. Right now only full manual mode is supported. JAX Array API #4440: Pull request #24894 synchronize by copybara-service bot
November 14, 2024 06:24 59s test_693048291
test_693048291
November 14, 2024 06:24 59s