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
1,873 workflow runs
1,873 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Pallas TPU] Fix boolean comparison JAX Array API #1847: Pull request #24226 synchronize by copybara-service bot
October 10, 2024 11:05 1m 1s test_684198399
test_684198399
October 10, 2024 11:05 1m 1s
[Pallas TPU] Fix boolean comparison JAX Array API #1846: Pull request #24226 synchronize by copybara-service bot
October 10, 2024 11:02 54s test_684198399
test_684198399
October 10, 2024 11:02 54s
[Pallas TPU] Fix boolean comparison JAX Array API #1845: Pull request #24226 opened by copybara-service bot
October 10, 2024 11:00 58s test_684198399
test_684198399
October 10, 2024 11:00 58s
[pallas] Added API docs for Triton and Mosaic GPU backends JAX Array API #1844: Pull request #24185 synchronize by superbobry
October 10, 2024 10:43 53s superbobry:docs
superbobry:docs
October 10, 2024 10:43 53s
[Mosaic] Parameterize the number of lanes and sublanes in TPU dialects. JAX Array API #1843: Pull request #24022 synchronize by copybara-service bot
October 10, 2024 10:41 59s test_680633619
test_680633619
October 10, 2024 10:41 59s
[jax2tf] Disable jax2tf with non-native serialization. JAX Array API #1842: Pull request #24183 synchronize by copybara-service bot
October 10, 2024 06:02 1m 2s test_683472001
test_683472001
October 10, 2024 06:02 1m 2s
[sharding_in_types] Add support for nary ops to propagate sharding when 1 input is sharded and all others are replicated. JAX Array API #1841: Pull request #24223 synchronize by copybara-service bot
October 10, 2024 04:24 59s test_684151551
test_684151551
October 10, 2024 04:24 59s
[sharding_in_types] Add support for nary ops to propagate sharding wh… JAX Array API #1840: Commit 351187d pushed by copybara-service bot
October 10, 2024 04:24 52s main
main
October 10, 2024 04:24 52s
[sharding_in_types] Add support for nary ops to propagate sharding when 1 input is sharded and all others are replicated. JAX Array API #1839: Pull request #24223 synchronize by copybara-service bot
October 10, 2024 04:09 56s test_684151551
test_684151551
October 10, 2024 04:09 56s
[sharding_in_types] Normalize partition specs when creating avals so that P(None, None) and P() are treated as replicated and equivalent. Shardings on avals are always normalized. JAX Array API #1838: Pull request #24224 synchronize by copybara-service bot
October 10, 2024 04:08 53s test_684285486
test_684285486
October 10, 2024 04:08 53s
[sharding_in_types] Normalize partition specs when creating avals so that P(None, None) and P() are treated as replicated and equivalent. Shardings on avals are always normalized. JAX Array API #1837: Pull request #24224 opened by copybara-service bot
October 10, 2024 04:07 55s test_684285486
test_684285486
October 10, 2024 04:07 55s
[sharding_in_types] Add support for nary ops to propagate sharding when 1 input is sharded and all others are replicated. JAX Array API #1836: Pull request #24223 opened by copybara-service bot
October 10, 2024 03:42 53s test_684151551
test_684151551
October 10, 2024 03:42 53s
[Pallas] Add tests for scalar elementwise operations JAX Array API #1835: Pull request #24222 synchronize by copybara-service bot
October 10, 2024 00:59 54s test_684227461
test_684227461
October 10, 2024 00:59 54s
[Pallas] Add tests for scalar elementwise operations JAX Array API #1834: Pull request #24222 synchronize by copybara-service bot
October 10, 2024 00:56 1m 0s test_684227461
test_684227461
October 10, 2024 00:56 1m 0s
[Pallas] Add tests for scalar elementwise operations JAX Array API #1833: Pull request #24222 opened by copybara-service bot
October 10, 2024 00:29 55s test_684227461
test_684227461
October 10, 2024 00:29 55s
Add Python 3.13.0 to JAX Docker images with CUDA 12.3 and CUDA 12.1. JAX Array API #1832: Commit b65be4e pushed by copybara-service bot
October 9, 2024 23:49 51s main
main
October 9, 2024 23:49 51s
Add Python 3.13.0 to JAX Docker images with CUDA 12.3 and CUDA 12.1. JAX Array API #1831: Pull request #24194 synchronize by copybara-service bot
October 9, 2024 23:40 52s test_683252345
test_683252345
October 9, 2024 23:40 52s
Add Python 3.13.0 to JAX Docker images with CUDA 12.3 and CUDA 12.1. JAX Array API #1830: Pull request #24194 synchronize by copybara-service bot
October 9, 2024 23:29 53s test_683252345
test_683252345
October 9, 2024 23:29 53s
Allow simple host call that uses host tensor as parameter/result. JAX Array API #1829: Pull request #24057 synchronize by copybara-service bot
October 9, 2024 23:10 58s test_680683100
test_680683100
October 9, 2024 23:10 58s
[Mosaic TPU] Support native int4 @ int4 JAX Array API #1828: Pull request #24220 opened by copybara-service bot
October 9, 2024 22:50 54s test_683883622
test_683883622
October 9, 2024 22:50 54s
Add Python 3.13.0 to JAX Docker images with CUDA 12.3 and CUDA 12.1. JAX Array API #1827: Pull request #24194 synchronize by copybara-service bot
October 9, 2024 22:19 56s test_683252345
test_683252345
October 9, 2024 22:19 56s
[Pallas] Add lowering for threefry PRNG. JAX Array API #1826: Pull request #23778 synchronize by copybara-service bot
October 9, 2024 21:48 54s test_663058435
test_663058435
October 9, 2024 21:48 54s
[Pallas] Add lowering for threefry PRNG. JAX Array API #1825: Commit 7341842 pushed by copybara-service bot
October 9, 2024 21:48 52s main
main
October 9, 2024 21:48 52s
Add float8_e4m3 and float8_e3m4 types support JAX Array API #1824: Pull request #23585 synchronize by apivovarov
October 9, 2024 21:36 Action required apivovarov:float8_e4m3
apivovarov:float8_e4m3
October 9, 2024 21:36 Action required
[Pallas] Add lowering for threefry PRNG. JAX Array API #1823: Pull request #23778 synchronize by copybara-service bot
October 9, 2024 21:13 54s test_663058435
test_663058435
October 9, 2024 21:13 54s