Skip to content

Actions: jax-ml/jax

Actions

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
ci-build.yaml
37,315 workflow runs
37,315 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Mosaic] Add sign_extended attribute to tpu::UnpackSubelementsOp. CI #98251: Pull request #32477 synchronize by copybara-service bot
8m 48s test_817185268
test_817185268
8m 48s
[Mosaic] Add sign_extended attribute to tpu::UnpackSubelementsOp. CI #98250: Commit fbf4610 pushed by copybara-service bot
9m 4s main
main
9m 4s
[Mosaic] Relax the restrictions of small second minor tiling reshape. CI #98249: Pull request #32518 synchronize by copybara-service bot
8m 26s test_817703467
test_817703467
8m 26s
Make eager shmap + check_vma=False + axis_index work with explicit mode. CI #98248: Commit 9855a2e pushed by copybara-service bot
8m 17s main
main
8m 17s
Make eager shmap + check_vma=False + axis_index work with explicit mode. CI #98247: Pull request #32540 synchronize by copybara-service bot
10m 33s test_818107715
test_818107715
10m 33s
Make eager shmap + check_vma=False + axis_index work with explicit mode. CI #98246: Pull request #32540 opened by copybara-service bot
9m 6s test_818107715
test_818107715
9m 6s
Disallow nesting with jax.experimental.shard_map. This means to remove the auto= parameter from the experimental endpoint. Please use jax.shard_map if you want to nest shard_map calls. CI #98245: Pull request #32531 synchronize by copybara-service bot
10m 39s test_817792376
test_817792376
10m 39s
Disallow nesting with jax.experimental.shard_map. This means to remov… CI #98244: Commit dfadef7 pushed by copybara-service bot
8m 46s main
main
8m 46s
Disallow nesting with jax.experimental.shard_map. This means to remove the auto= parameter from the experimental endpoint. Please use jax.shard_map if you want to nest shard_map calls. CI #98243: Pull request #32531 synchronize by copybara-service bot
1m 55s test_817792376
test_817792376
1m 55s
Disallow nesting with jax.experimental.shard_map. This means to remove the auto= parameter from the experimental endpoint. Please use jax.shard_map if you want to nest shard_map calls. CI #98242: Pull request #32531 synchronize by copybara-service bot
8m 45s test_817792376
test_817792376
8m 45s
Deprecate jax.experimental.shard_map CI #98241: Pull request #32533 synchronize by jakevdp
8m 42s jakevdp:dep-shard-map
jakevdp:dep-shard-map
8m 42s
fix(docs): regularized incomplete beta function (jax.scipy.special.betainc) formula corrected in docstring CI #98240: Pull request #32539 opened by Qazalbash
Action required Qazalbash:fix-betainc-doc
Qazalbash:fix-betainc-doc
Action required
Automated Code Change CI #98239: Pull request #32538 opened by copybara-service bot
8m 10s test_817846083
test_817846083
8m 10s
Update XLA dependency to use revision http://github.com/openxla/xla/c… CI #98238: Commit bae1fbd pushed by copybara-service bot
9m 11s main
main
9m 11s
Add a fancy transpose rule for reshard_p so that we can do the accum under the right context on the backward pass which is the source's context. CI #98237: Pull request #32537 synchronize by copybara-service bot
8m 42s test_817897983
test_817897983
8m 42s
Add a fancy transpose rule for reshard_p so that we can do the accu… CI #98236: Commit c755701 pushed by copybara-service bot
8m 57s main
main
8m 57s
Add a fancy transpose rule for reshard_p so that we can do the accum under the right context on the backward pass which is the source's context. CI #98235: Pull request #32537 synchronize by copybara-service bot
8m 59s test_817897983
test_817897983
8m 59s
Add a fancy transpose rule for reshard_p so that we can do the accum under the right context on the backward pass which is the source's context. CI #98234: Pull request #32537 synchronize by copybara-service bot
6m 17s test_817897983
test_817897983
6m 17s
Add a fancy transpose rule for reshard_p so that we can do the accum under the right context on the backward pass which is the source's context. CI #98233: Pull request #32537 opened by copybara-service bot
2m 27s test_817897983
test_817897983
2m 27s
[JAX] Make Colocated Python object destruction use a sorted device list CI #98232: Commit 5cdc116 pushed by copybara-service bot
9m 44s main
main
9m 44s
Merge pull request #32535 from jakevdp:pallas-shard-map CI #98231: Commit 16a8ca4 pushed by copybara-service bot
1m 45s main
main
1m 45s
[JAX] Make Colocated Python object destruction use a sorted device list CI #98230: Pull request #32536 synchronize by copybara-service bot
8m 59s test_817820409
test_817820409
8m 59s
[IFRT] Migrate Array::pjrt_layout() callers to interpret nullptr as a default layout CI #98229: Pull request #32520 synchronize by copybara-service bot
8m 50s test_817466915
test_817466915
8m 50s
[IFRT] Migrate Array::pjrt_layout() callers to interpret nullptr … CI #98228: Commit e64fe93 pushed by copybara-service bot
8m 51s main
main
8m 51s
[IFRT] Migrate Array::pjrt_layout() callers to interpret nullptr as a default layout CI #98227: Pull request #32520 synchronize by copybara-service bot
9m 13s test_817466915
test_817466915
9m 13s