Skip to content

Actions: jax-ml/jax

Actions

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
ci-build.yaml
27,900 workflow runs
27,900 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[mosaic_gpu] The profiler now uses FFI calls for creating events and computing elapsed time CI #65633: Pull request #24755 opened by copybara-service bot
November 7, 2024 09:05 10m 42s test_693707186
test_693707186
November 7, 2024 09:05 10m 42s
[Mosaic] Fix canonicalize_extract op name. CI #65632: Pull request #24741 synchronize by copybara-service bot
November 7, 2024 08:04 10m 45s test_693747564
test_693747564
November 7, 2024 08:04 10m 45s
[Mosaic] Fix handling of i1 splat constants CI #65631: Pull request #24735 synchronize by copybara-service bot
November 7, 2024 06:24 10m 53s test_693521666
test_693521666
November 7, 2024 06:24 10m 53s
[XLA] Make our LLVM usage more googley CI #65630: Pull request #24706 synchronize by copybara-service bot
November 7, 2024 05:23 11m 3s test_688333737
test_688333737
November 7, 2024 05:23 11m 3s
Propagate CopySemantics from python to C++ transfer APIs so that device_put works correctly in presence of copy/donate options that user specified. CI #65629: Pull request #24752 opened by copybara-service bot
November 7, 2024 03:39 11m 14s test_693801738
test_693801738
November 7, 2024 03:39 11m 14s
Allow setting default_device with platform names. CI #65628: Pull request #24751 opened by Stella-S-Yan
November 7, 2024 00:36 Action required Stella-S-Yan:feature/default_device_str
Stella-S-Yan:feature/default_device_str
November 7, 2024 00:36 Action required
Determine symmetric linear op fro CG from abstract output dtype CI #65627: Pull request #23486 synchronize by Joshuaalbert
November 7, 2024 00:17 Action required Joshuaalbert:main
Joshuaalbert:main
November 7, 2024 00:17 Action required
disable remat HLO pass by default CI #65626: Pull request #24750 opened by keshavb96
November 7, 2024 00:13 Action required keshavb96:disable_default_remat_pass
keshavb96:disable_default_remat_pass
November 7, 2024 00:13 Action required
[SDY] add JAX lowering to Shardy ShardingGroupOp for shard_alike. CI #65625: Pull request #24724 synchronize by copybara-service bot
November 6, 2024 23:20 9m 30s test_688225820
test_688225820
November 6, 2024 23:20 9m 30s
Re-factor build CLI to a subcommand based approach CI #65624: Pull request #24695 synchronize by copybara-service bot
November 6, 2024 23:15 11m 20s test_691466647
test_691466647
November 6, 2024 23:15 11m 20s
Re-factor build CLI to a subcommand based approach CI #65623: Pull request #24695 synchronize by copybara-service bot
November 6, 2024 22:59 9m 53s test_691466647
test_691466647
November 6, 2024 22:59 9m 53s
RFC: specify jit static args via Static annotation CI #65622: Pull request #24705 synchronize by jakevdp
November 6, 2024 21:55 10m 1s jakevdp:jit-static
jakevdp:jit-static
November 6, 2024 21:55 10m 1s
[Pallas TPU] Lower lax.exp2 to MLIR math.exp2 directly CI #65621: Pull request #24746 synchronize by copybara-service bot
November 6, 2024 21:47 9m 42s test_693772684
test_693772684
November 6, 2024 21:47 9m 42s
[Pallas TPU] Lower lax.exp2 to MLIR math.exp2 directly CI #65620: Pull request #24746 synchronize by copybara-service bot
November 6, 2024 21:41 6m 33s test_693772684
test_693772684
November 6, 2024 21:41 6m 33s
Merge pull request #24602 from rdyro:rdyro-decode-attention-mask CI #65619: Commit 37af100 pushed by copybara-service bot
November 6, 2024 21:05 10m 36s main
main
November 6, 2024 21:05 10m 36s
Plumb dot dimension numbers into TPU matmul op. CI #65618: Pull request #24561 synchronize by copybara-service bot
November 6, 2024 21:01 8m 59s test_689442832
test_689442832
November 6, 2024 21:01 8m 59s
When caching is enabled, also enable XLA caching features as well CI #65617: Pull request #22899 synchronize by trevor-m
November 6, 2024 20:57 9m 26s trevor-m:cache
trevor-m:cache
November 6, 2024 20:57 9m 26s
jnp.reshape: raise TypeError when specifying newshape CI #65616: Pull request #24748 opened by jakevdp
November 6, 2024 20:57 8m 18s jakevdp:reshape-dep
jakevdp:reshape-dep
November 6, 2024 20:57 8m 18s
Re-factor build CLI to a subcommand based approach CI #65615: Pull request #24695 synchronize by copybara-service bot
November 6, 2024 20:49 11m 15s test_691466647
test_691466647
November 6, 2024 20:49 11m 15s
Add a GPU implementation of lax.linalg.eig. CI #65614: Pull request #24663 synchronize by copybara-service bot
November 6, 2024 20:49 9m 17s test_691072237
test_691072237
November 6, 2024 20:49 9m 17s
Add a GPU implementation of lax.linalg.eig. CI #65613: Pull request #24663 synchronize by copybara-service bot
November 6, 2024 20:39 9m 14s test_691072237
test_691072237
November 6, 2024 20:39 9m 14s
[SDY] add JAX lowering to Shardy ShardingGroupOp for shard_alike. CI #65612: Pull request #24724 synchronize by copybara-service bot
November 6, 2024 20:35 8m 13s test_688225820
test_688225820
November 6, 2024 20:35 8m 13s
Update XLA dependency to use revision CI #65611: Commit dc33a28 pushed by copybara-service bot
November 6, 2024 20:19 8m 29s main
main
November 6, 2024 20:19 8m 29s
More thorough propagation of host linear layout CI #65610: Pull request #24734 synchronize by copybara-service bot
November 6, 2024 19:50 9m 0s test_693505340
test_693505340
November 6, 2024 19:50 9m 0s
register_dataclass: allow marking static fields via dataclass.field CI #65609: Pull request #24700 synchronize by jakevdp
November 6, 2024 19:18 10m 12s jakevdp:register-dataclass
jakevdp:register-dataclass
November 6, 2024 19:18 10m 12s