Skip to content

Actions: jax-ml/jax

Actions

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
ci-build.yaml
25,924 workflow runs
25,924 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Removed unused imports in jax.experimental.mosaic.gpu.core CI #62550: Pull request #23966 opened by copybara-service bot
September 27, 2024 08:49 7m 17s test_679289036
test_679289036
September 27, 2024 08:49 7m 17s
[Pallas TPU] Add a note in the Pallas Quickstart documentation about … CI #62549: Commit ab4590c pushed by copybara-service bot
September 27, 2024 08:33 10m 46s main
main
September 27, 2024 08:33 10m 46s
[Pallas TPU] Add a note in the Pallas Quickstart documentation about the instructions of running the existing example on TPU CI #62548: Pull request #23889 synchronize by copybara-service bot
September 27, 2024 08:08 9m 29s test_678420788
test_678420788
September 27, 2024 08:08 9m 29s
Enable weight offloading tests that are supported on GPUs now CI #62547: Pull request #23965 opened by zhenying-liu
September 27, 2024 06:46 9m 35s zhenying-liu:weight-offloading-test
zhenying-liu:weight-offloading-test
September 27, 2024 06:46 9m 35s
Update device_memory_profiling.md CI #62546: Pull request #23964 opened by garymm
September 27, 2024 05:10 Action required garymm:patch-2
garymm:patch-2
September 27, 2024 05:10 Action required
Add float8_e4m3 and float8_e3m4 types support CI #62545: Pull request #23585 synchronize by apivovarov
September 27, 2024 03:41 Action required apivovarov:float8_e4m3
apivovarov:float8_e4m3
September 27, 2024 03:41 Action required
Add float8_e4m3 and float8_e3m4 types support CI #62544: Pull request #23585 synchronize by apivovarov
September 27, 2024 03:39 Action required apivovarov:float8_e4m3
apivovarov:float8_e4m3
September 27, 2024 03:39 Action required
Add float8_e4m3 and float8_e3m4 types support CI #62543: Pull request #23585 synchronize by apivovarov
September 27, 2024 03:38 Action required apivovarov:float8_e4m3
apivovarov:float8_e4m3
September 27, 2024 03:38 Action required
Adding JAX_LOGGING_LEVEL configuration option CI #62542: Pull request #23468 synchronize by rdyro
September 27, 2024 01:58 10m 0s rdyro:rdyro-add-logging-env
rdyro:rdyro-add-logging-env
September 27, 2024 01:58 10m 0s
Merge pull request #23853 from zhenying-liu:remat-scan CI #62541: Commit 5a1549c pushed by copybara-service bot
September 27, 2024 01:12 9m 18s main
main
September 27, 2024 01:12 9m 18s
Remove pylint sections from pyproject.toml (use ruff instead) CI #62540: Pull request #23961 synchronize by apivovarov
September 26, 2024 23:30 Action required apivovarov:pylint2
apivovarov:pylint2
September 26, 2024 23:30 Action required
Remove pylint sections from pyproject.toml (use ruff instead) CI #62539: Pull request #23961 opened by apivovarov
September 26, 2024 23:18 Action required apivovarov:pylint2
apivovarov:pylint2
September 26, 2024 23:18 Action required
Add very simple batching support for ragged_dot. CI #62538: Pull request #23685 synchronize by copybara-service bot
September 26, 2024 23:15 11m 46s test_675407251
test_675407251
September 26, 2024 23:15 11m 46s
[Pallas] Simplify OpsExtraTest by skipping 64-bit tests on 32-bit environments CI #62537: Pull request #23960 opened by copybara-service bot
September 26, 2024 22:50 10m 31s test_679307748
test_679307748
September 26, 2024 22:50 10m 31s
[Pallas TPU] Add a note in the Pallas Quickstart documentation about the instructions of running the existing example on TPU CI #62536: Pull request #23889 synchronize by copybara-service bot
September 26, 2024 22:48 10m 34s test_678420788
test_678420788
September 26, 2024 22:48 10m 34s
[Pallas TPU] Add a note in the Pallas Quickstart documentation about the instructions of running the existing example on TPU CI #62534: Pull request #23889 synchronize by copybara-service bot
September 26, 2024 22:41 7m 21s test_678420788
test_678420788
September 26, 2024 22:41 7m 21s
Add .pylintrc file CI #62533: Pull request #23959 opened by apivovarov
September 26, 2024 22:18 Action required apivovarov:pylintrc
apivovarov:pylintrc
September 26, 2024 22:18 Action required
[Pallas TPU] Add a note in the Pallas Quickstart documentation about the instructions of running the existing example on TPU CI #62532: Pull request #23889 synchronize by copybara-service bot
September 26, 2024 21:51 10m 58s test_678420788
test_678420788
September 26, 2024 21:51 10m 58s
[Pallas TPU] Add a note in the Pallas Quickstart documentation about the instructions of running the existing example on TPU CI #62531: Pull request #23889 synchronize by copybara-service bot
September 26, 2024 21:48 3m 47s test_678420788
test_678420788
September 26, 2024 21:48 3m 47s
Better documentation for jnp.choose CI #62530: Pull request #23957 opened by jakevdp
September 26, 2024 21:16 8m 55s jakevdp:choice-doc
jakevdp:choice-doc
September 26, 2024 21:16 8m 55s
[XLA:Mosaic][Pallas] Enable vector.ExtractOp for non-zero indices. CI #62529: Commit 9f4e8d0 pushed by copybara-service bot
September 26, 2024 20:57 9m 34s main
main
September 26, 2024 20:57 9m 34s
Rewrite mosaic concat to support operand shapes that do not align with native shapes, Expand tests to cover multi operand, batch dim concat, etc. CI #62528: Pull request #23714 synchronize by copybara-service bot
September 26, 2024 20:18 9m 4s test_675835891
test_675835891
September 26, 2024 20:18 9m 4s
Merge pull request #23949 from rajasekharporeddy:testbranch1 CI #62527: Commit 46dbb65 pushed by copybara-service bot
September 26, 2024 20:09 10m 21s main
main
September 26, 2024 20:09 10m 21s
Exposed sequential iteration index via pl.program_id in Pallas Mosaic GPU CI #62526: Pull request #23956 opened by copybara-service bot
September 26, 2024 20:02 10m 53s test_679261898
test_679261898
September 26, 2024 20:02 10m 53s
Fixed Pallas Mosaic GPU test following recent changes CI #62525: Pull request #23955 opened by copybara-service bot
September 26, 2024 20:02 9m 4s test_679261897
test_679261897
September 26, 2024 20:02 9m 4s