Skip to content

Actions: jax-ml/jax

Actions

CI - Bazel CPU tests (RBE)

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
bazel_cpu_rbe.yml
2,004 workflow runs
2,004 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Add an experimental interface for customizing DCE behavior. CI - Bazel CPU tests (RBE) #2029: Pull request #25956 synchronize by copybara-service bot
January 19, 2025 01:53 5m 45s test_716596154
test_716596154
January 19, 2025 01:53 5m 45s
Add an experimental interface for customizing DCE behavior. CI - Bazel CPU tests (RBE) #2028: Pull request #25956 synchronize by copybara-service bot
January 19, 2025 01:45 5m 50s test_716596154
test_716596154
January 19, 2025 01:45 5m 50s
Enable pivoted QR on GPU via MAGMA. CI - Bazel CPU tests (RBE) #2027: Pull request #25955 synchronize by tttc3
January 18, 2025 10:07 Action required tttc3:magma_qr
tttc3:magma_qr
January 18, 2025 10:07 Action required
Enable pivoted QR on GPU via MAGMA. CI - Bazel CPU tests (RBE) #2026: Pull request #25955 synchronize by tttc3
January 18, 2025 10:07 Action required tttc3:magma_qr
tttc3:magma_qr
January 18, 2025 10:07 Action required
Add test case for jit compile of vmap on gpu CI - Bazel CPU tests (RBE) #2025: Pull request #25980 opened by codinglover222
January 18, 2025 08:16 Action required codinglover222:jit-vmap-compile-test
codinglover222:jit-vmap-compile-test
January 18, 2025 08:16 Action required
Remove memories flag now that JAX 0.5.0 has been released since it always defaults to True. CI - Bazel CPU tests (RBE) #2024: Pull request #25979 synchronize by copybara-service bot
January 18, 2025 06:13 4m 19s test_716903141
test_716903141
January 18, 2025 06:13 4m 19s
Remove memories flag now that JAX 0.5.0 has been released since it always defaults to True. CI - Bazel CPU tests (RBE) #2023: Pull request #25979 synchronize by copybara-service bot
January 18, 2025 05:59 3m 56s test_716903141
test_716903141
January 18, 2025 05:59 3m 56s
Remove memories flag now that JAX 0.5.0 has been released since it always defaults to True. CI - Bazel CPU tests (RBE) #2022: Pull request #25979 opened by copybara-service bot
January 18, 2025 05:48 10m 28s test_716903141
test_716903141
January 18, 2025 05:48 10m 28s
Add reshard API in experimental. Currently for sharding_in_types we have 2 APIs: mesh_cast and reshard. Both work in sharding_in_types mode and affect the sharding of the aval. Following are the semantics of both: CI - Bazel CPU tests (RBE) #2021: Pull request #25978 opened by copybara-service bot
January 18, 2025 04:34 6m 7s test_716879595
test_716879595
January 18, 2025 04:34 6m 7s
Adds a minimal but viable implementation of string arrays (with numpy.dtypes.StringDType) in JAX. Currently this only supports making of a string array by means of either jax.numpy.asarray or jax.device_put and reading it back with jax.device_get. CI - Bazel CPU tests (RBE) #2020: Pull request #25918 synchronize by copybara-service bot
January 18, 2025 01:45 5m 17s test_716042460
test_716042460
January 18, 2025 01:45 5m 17s
fix aliased arg in async pallas docs CI - Bazel CPU tests (RBE) #2019: Pull request #25977 opened by arvoelke
January 18, 2025 01:32 Action required arvoelke:fix-async-pallas-doc
arvoelke:fix-async-pallas-doc
January 18, 2025 01:32 Action required
Add a sharding rule for reduce_precision_p and properly thread eqn.ctx in loops.py where we create pe.new_jaxpr_eqn's CI - Bazel CPU tests (RBE) #2018: Pull request #25974 synchronize by copybara-service bot
January 18, 2025 01:31 4m 0s test_716841435
test_716841435
January 18, 2025 01:31 4m 0s
add f-string to mosaic memory space error msg CI - Bazel CPU tests (RBE) #2017: Pull request #25976 opened by arvoelke
January 18, 2025 01:28 9m 16s arvoelke:fix-memory-space-error
arvoelke:fix-memory-space-error
January 18, 2025 01:28 9m 16s
Add a sharding rule for reduce_precision_p and properly thread eqn.ctx in loops.py where we create pe.new_jaxpr_eqn's CI - Bazel CPU tests (RBE) #2016: Pull request #25974 synchronize by copybara-service bot
January 18, 2025 01:16 4m 35s test_716841435
test_716841435
January 18, 2025 01:16 4m 35s
Remove unnecessary use of xla_client.OpMetadata class. CI - Bazel CPU tests (RBE) #2015: Pull request #25975 opened by copybara-service bot
January 18, 2025 01:15 3m 52s test_716844471
test_716844471
January 18, 2025 01:15 3m 52s
Remove CUDA rpaths from jaxlib build. CI - Bazel CPU tests (RBE) #2014: Pull request #25973 synchronize by copybara-service bot
January 18, 2025 01:09 3m 53s test_716839689
test_716839689
January 18, 2025 01:09 3m 53s
Add a sharding rule for reduce_precision_p and properly thread eqn.ctx in loops.py where we create pe.new_jaxpr_eqn's CI - Bazel CPU tests (RBE) #2013: Pull request #25974 opened by copybara-service bot
January 18, 2025 01:06 4m 34s test_716841435
test_716841435
January 18, 2025 01:06 4m 34s
Remove CUDA rpaths from jaxlib build. CI - Bazel CPU tests (RBE) #2012: Pull request #25973 synchronize by copybara-service bot
January 18, 2025 00:56 3m 59s test_716839689
test_716839689
January 18, 2025 00:56 3m 59s
Remove CUDA rpaths from jaxlib build. CI - Bazel CPU tests (RBE) #2011: Pull request #25973 opened by copybara-service bot
January 18, 2025 00:54 2m 9s test_716839689
test_716839689
January 18, 2025 00:54 2m 9s
Expose hidden_axes via jax namespace as public API. Also mention it as a workaround for primitives we don't support yet. CI - Bazel CPU tests (RBE) #2010: Pull request #25970 synchronize by copybara-service bot
January 18, 2025 00:49 3m 58s test_716770575
test_716770575
January 18, 2025 00:49 3m 58s
Adds a minimal but viable implementation of string arrays (with numpy.dtypes.StringDType) in JAX. Currently this only supports making of a string array by means of either jax.numpy.asarray or jax.device_put and reading it back with jax.device_get. CI - Bazel CPU tests (RBE) #2009: Pull request #25918 synchronize by copybara-service bot
January 18, 2025 00:46 4m 56s test_716042460
test_716042460
January 18, 2025 00:46 4m 56s
Expose hidden_axes via jax namespace as public API. Also mention it as a workaround for primitives we don't support yet. CI - Bazel CPU tests (RBE) #2008: Pull request #25970 synchronize by copybara-service bot
January 18, 2025 00:33 5m 40s test_716770575
test_716770575
January 18, 2025 00:33 5m 40s
Add GitHub action workflow for Bazel CUDA continuous tests CI - Bazel CPU tests (RBE) #2007: Pull request #25717 synchronize by copybara-service bot
January 18, 2025 00:29 6m 25s test_705238265
test_705238265
January 18, 2025 00:29 6m 25s
Support Hessian of gamma-distributed samples CI - Bazel CPU tests (RBE) #2006: Pull request #21432 synchronize by NeilGirdhar
January 18, 2025 00:15 Action required NeilGirdhar:gamma_hessian
NeilGirdhar:gamma_hessian
January 18, 2025 00:15 Action required
Adds a minimal but viable implementation of string arrays (with numpy.dtypes.StringDType) in JAX. Currently this only supports making of a string array by means of either jax.numpy.asarray or jax.device_put and reading it back with jax.device_get. CI - Bazel CPU tests (RBE) #2005: Pull request #25918 synchronize by copybara-service bot
January 17, 2025 23:30 4m 19s test_716042460
test_716042460
January 17, 2025 23:30 4m 19s