Skip to content

Actions: ROCm/jax

Actions

JAX Array API

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
jax-array-api.yml
105 workflow runs
105 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

fix binomial sample test on TPU JAX Array API #105: Commit 58a202a pushed by charleshofer
February 4, 2025 16:58 3m 7s main
main
February 4, 2025 16:58 3m 7s
Merge pull request #25812 from Cjkkkk:segment_ids JAX Array API #104: Commit 7164c6b pushed by charleshofer
February 3, 2025 16:07 2m 54s main
main
February 3, 2025 16:07 2m 54s
Re-enable Pallas bf16 exp2 tests on TPU JAX Array API #103: Commit c1e1360 pushed by charleshofer
January 31, 2025 16:56 3m 5s main
main
January 31, 2025 16:56 3m 5s
Disable pytorch_interoperability_test under asan. JAX Array API #102: Commit 42fd586 pushed by charleshofer
January 27, 2025 17:07 3m 10s main
main
January 27, 2025 17:07 3m 10s
[sharding_in_types] Handle ShapeDtypeStruct inputs with sharding_in_t… JAX Array API #101: Commit c72ed26 pushed by charleshofer
January 14, 2025 16:11 3m 14s main
main
January 14, 2025 16:11 3m 14s
Update XLA dependency to use revision JAX Array API #100: Commit 5511949 pushed by charleshofer
January 8, 2025 17:43 3m 17s main
main
January 8, 2025 17:43 3m 17s
Add a register_custom_type_id function to the GPU plugins. JAX Array API #99: Commit a7f384c pushed by charleshofer
January 7, 2025 16:23 3m 28s main
main
January 7, 2025 16:23 3m 28s
Merge pull request #25516 from jakevdp:jep-effver JAX Array API #98: Commit 49c9246 pushed by charleshofer
December 19, 2024 16:34 3m 25s main
main
December 19, 2024 16:34 3m 25s
Merge pull request #25426 from hawkinsp:tls JAX Array API #97: Commit 99d675a pushed by charleshofer
December 12, 2024 17:35 3m 32s main
main
December 12, 2024 17:35 3m 32s
Merge pull request #25369 from jax-ml:mutable-arrays-ad JAX Array API #96: Commit 263d4d1 pushed by charleshofer
December 10, 2024 15:43 3m 40s main
main
December 10, 2024 15:43 3m 40s
Remove dead code after minimum jaxlib version bump to v0.4.36. JAX Array API #95: Commit 79318a0 pushed by charleshofer
December 9, 2024 16:22 1m 31s main
main
December 9, 2024 16:22 1m 31s
[Pallas:MGPU] Fix an overly strict precision requirement in tests JAX Array API #94: Commit aff7714 pushed by charleshofer
December 2, 2024 15:50 1m 10s main
main
December 2, 2024 15:50 1m 10s
Fix a weird interaction with set_local and empty tuples passed to it. JAX Array API #93: Commit 6763fcf pushed by charleshofer
November 26, 2024 21:23 1m 2s main
main
November 26, 2024 21:23 1m 2s
Charlie test PR JAX Array API #92: Pull request #154 opened by charleshofer
November 26, 2024 16:24 58s ci_fix_wheelhouse_relative_paths
ci_fix_wheelhouse_relative_paths
November 26, 2024 16:24 58s
[ROCm] fix jax and wheelhouse relative paths JAX Array API #91: Pull request #153 opened by Ruturaj4
November 26, 2024 15:44 1m 1s ci_fix_wheelhouse_relative_paths
ci_fix_wheelhouse_relative_paths
November 26, 2024 15:44 1m 1s
Test PR for upstream CI changes JAX Array API #90: Pull request #149 opened by charleshofer
November 22, 2024 18:57 1m 2s rocm-main
rocm-main
November 22, 2024 18:57 1m 2s
Merge pull request #24751 from Stella-S-Yan:feature/default_device_str JAX Array API #89: Commit cea8176 pushed by charleshofer
November 14, 2024 18:09 57s main
main
November 14, 2024 18:09 57s
[pallas:mosaic_gpu] Ported two pipelining optimizations to `emit_pipe… JAX Array API #88: Commit aefe621 pushed by charleshofer
November 14, 2024 16:53 59s main
main
November 14, 2024 16:53 59s
Remove GPU test with unreasonably large memory footprint. JAX Array API #87: Commit a99ccd9 pushed by charleshofer
November 12, 2024 16:53 1m 9s main
main
November 12, 2024 16:53 1m 9s
[Pallas:TPU] Fix some stale/wrong skip conditions. JAX Array API #86: Commit b6f5c95 pushed by charleshofer
November 6, 2024 18:59 1m 1s main
main
November 6, 2024 18:59 1m 1s
Merge branch 'jax-ml:main' into main JAX Array API #85: Commit e5a4556 pushed by charleshofer
November 4, 2024 23:26 56s main
main
November 4, 2024 23:26 56s
Merge branch 'rocm-main' into main JAX Array API #84: Commit 3d063fa pushed by charleshofer
November 4, 2024 23:19 55s main
main
November 4, 2024 23:19 55s
Remove the initial argument to jax.nn.softmax and `jax.nn.log_sof… JAX Array API #83: Commit e9acaa8 pushed by charleshofer
November 4, 2024 19:03 1m 0s main
main
November 4, 2024 19:03 1m 0s
Test CLI PR JAX Array API #82: Pull request #118 opened by charleshofer
October 31, 2024 20:16 1m 4s rocm-main
rocm-main
October 31, 2024 20:16 1m 4s
[Mosaic] Fix some imports. JAX Array API #81: Commit 7d9f565 pushed by charleshofer
October 31, 2024 17:04 1m 1s main
main
October 31, 2024 17:04 1m 1s