Skip to content

Actions: jax-ml/jax

Actions

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

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

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Remove memories flag now that JAX 0.5.0 has been released since it al… CI #70431: Commit 5a068da pushed by copybara-service bot
January 18, 2025 06:13 11m 4s main
main
January 18, 2025 06:13 11m 4s
Remove memories flag now that JAX 0.5.0 has been released since it always defaults to True. CI #70430: Pull request #25979 synchronize by copybara-service bot
January 18, 2025 05:59 11m 25s test_716903141
test_716903141
January 18, 2025 05:59 11m 25s
Remove memories flag now that JAX 0.5.0 has been released since it always defaults to True. CI #70429: Pull request #25979 opened by copybara-service bot
January 18, 2025 05:48 10m 54s test_716903141
test_716903141
January 18, 2025 05:48 10m 54s
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 #70428: Pull request #25978 opened by copybara-service bot
January 18, 2025 04:34 11m 57s test_716879595
test_716879595
January 18, 2025 04:34 11m 57s
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 #70427: Pull request #25918 synchronize by copybara-service bot
January 18, 2025 01:45 11m 38s test_716042460
test_716042460
January 18, 2025 01:45 11m 38s
fix aliased arg in async pallas docs CI #70426: 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 #70425: Pull request #25974 synchronize by copybara-service bot
January 18, 2025 01:31 11m 35s test_716841435
test_716841435
January 18, 2025 01:31 11m 35s
Add a sharding rule for reduce_precision_p and properly thread eqn.… CI #70424: Commit 36daf36 pushed by copybara-service bot
January 18, 2025 01:31 11m 31s main
main
January 18, 2025 01:31 11m 31s
add f-string to mosaic memory space error msg CI #70423: Pull request #25976 opened by arvoelke
January 18, 2025 01:28 11m 2s arvoelke:fix-memory-space-error
arvoelke:fix-memory-space-error
January 18, 2025 01:28 11m 2s
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 #70422: Pull request #25974 synchronize by copybara-service bot
January 18, 2025 01:16 12m 2s test_716841435
test_716841435
January 18, 2025 01:16 12m 2s
Remove unnecessary use of xla_client.OpMetadata class. CI #70421: Pull request #25975 opened by copybara-service bot
January 18, 2025 01:15 11m 15s test_716844471
test_716844471
January 18, 2025 01:15 11m 15s
Remove CUDA rpaths from jaxlib build. CI #70420: Pull request #25973 synchronize by copybara-service bot
January 18, 2025 01:09 12m 34s test_716839689
test_716839689
January 18, 2025 01:09 12m 34s
Remove CUDA rpaths from jaxlib build. CI #70419: Commit 034e967 pushed by copybara-service bot
January 18, 2025 01:09 11m 45s main
main
January 18, 2025 01:09 11m 45s
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 #70418: Pull request #25974 opened by copybara-service bot
January 18, 2025 01:06 10m 3s test_716841435
test_716841435
January 18, 2025 01:06 10m 3s
Remove CUDA rpaths from jaxlib build. CI #70417: Pull request #25973 synchronize by copybara-service bot
January 18, 2025 00:56 11m 51s test_716839689
test_716839689
January 18, 2025 00:56 11m 51s
Remove CUDA rpaths from jaxlib build. CI #70416: Pull request #25973 opened by copybara-service bot
January 18, 2025 00:54 2m 10s test_716839689
test_716839689
January 18, 2025 00:54 2m 10s
Expose hidden_axes via jax namespace as public API. Also mention it as a workaround for primitives we don't support yet. CI #70415: Pull request #25970 synchronize by copybara-service bot
January 18, 2025 00:49 11m 35s test_716770575
test_716770575
January 18, 2025 00:49 11m 35s
Expose hidden_axes via jax namespace as public API. Also mention it a… CI #70414: Commit c7f8d17 pushed by copybara-service bot
January 18, 2025 00:49 11m 43s main
main
January 18, 2025 00:49 11m 43s
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 #70413: Pull request #25918 synchronize by copybara-service bot
January 18, 2025 00:46 8m 6s test_716042460
test_716042460
January 18, 2025 00:46 8m 6s
Expose hidden_axes via jax namespace as public API. Also mention it as a workaround for primitives we don't support yet. CI #70412: Pull request #25970 synchronize by copybara-service bot
January 18, 2025 00:33 11m 17s test_716770575
test_716770575
January 18, 2025 00:33 11m 17s
Add GitHub action workflow for Bazel CUDA continuous tests CI #70411: Pull request #25717 synchronize by copybara-service bot
January 18, 2025 00:29 12m 3s test_705238265
test_705238265
January 18, 2025 00:29 12m 3s
Support Hessian of gamma-distributed samples CI #70410: 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 #70409: Pull request #25918 synchronize by copybara-service bot
January 17, 2025 23:30 9m 3s test_716042460
test_716042460
January 17, 2025 23:30 9m 3s
internal: check integer overflow in lax.asarray CI #70408: Pull request #25969 synchronize by jakevdp
January 17, 2025 22:38 11m 32s jakevdp:fix-util
jakevdp:fix-util
January 17, 2025 22:38 11m 32s
Refactor JAX wheel build rules to control the wheel filename and maintain reproducible wheel content and filename results. CI #70407: Pull request #25126 synchronize by copybara-service bot
January 17, 2025 22:33 12m 21s test_699315679
test_699315679
January 17, 2025 22:33 12m 21s