Skip to content

Actions: jax-ml/jax

Actions

All workflows

Actions

Loading...
Loading

Showing runs from all workflows
52,763 workflow runs
52,763 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Add Scaled Dot Product Attention for FP8 CI #64228: Pull request #22670 synchronize by wenscarl
October 18, 2024 18:29 Action required wenscarl:fp8_sdpa
wenscarl:fp8_sdpa
October 18, 2024 18:29 Action required
Document jax config to disable remat HLO pass CI #64227: Pull request #24391 opened by keshavb96
October 18, 2024 18:28 Action required keshavb96:remat_documentation
keshavb96:remat_documentation
October 18, 2024 18:28 Action required
Document jax config to disable remat HLO pass JAX Array API #2476: Pull request #24391 opened by keshavb96
October 18, 2024 18:28 Action required keshavb96:remat_documentation
keshavb96:remat_documentation
October 18, 2024 18:28 Action required
Add Scaled Dot Product Attention for FP8 JAX Array API #2475: Pull request #22670 synchronize by wenscarl
October 18, 2024 18:22 Action required wenscarl:fp8_sdpa
wenscarl:fp8_sdpa
October 18, 2024 18:22 Action required
Add Scaled Dot Product Attention for FP8 CI #64226: Pull request #22670 synchronize by wenscarl
October 18, 2024 18:22 Action required wenscarl:fp8_sdpa
wenscarl:fp8_sdpa
October 18, 2024 18:22 Action required
PR #24388 CodeQL #19728: by copybara-service bot
October 18, 2024 17:35 4m 14s
October 18, 2024 17:35 4m 14s
[Pallas TPU] Use new MLIR op names CI #64225: Pull request #24388 synchronize by copybara-service bot
October 18, 2024 17:35 9m 57s test_687278048
test_687278048
October 18, 2024 17:35 9m 57s
[Pallas TPU] Use new MLIR op names JAX Array API #2474: Pull request #24388 synchronize by copybara-service bot
October 18, 2024 17:35 58s test_687278048
test_687278048
October 18, 2024 17:35 58s
[Pallas:MGPU] Fix the implementation of WGMMA with transposed RHS CI #64224: Commit bbcc3ee pushed by copybara-service bot
October 18, 2024 17:31 11m 27s main
main
October 18, 2024 17:31 11m 27s
[Pallas:MGPU] Fix the implementation of WGMMA with transposed RHS JAX Array API #2473: Commit bbcc3ee pushed by copybara-service bot
October 18, 2024 17:31 54s main
main
October 18, 2024 17:31 54s
Push on main CodeQL #19727: by copybara-service bot
October 18, 2024 17:31 4m 27s
October 18, 2024 17:31 4m 27s
[sharding_in_types] If out_aval.sharding is not None and the user specified out_sharding is None, concretize it with the device assignment available and add it to the final out_shardings that's used for lowering and compilation. CI #64223: Pull request #24387 synchronize by copybara-service bot
October 18, 2024 17:28 11m 40s test_687174276
test_687174276
October 18, 2024 17:28 11m 40s
[sharding_in_types] If out_aval.sharding is not None and the user specified out_sharding is None, concretize it with the device assignment available and add it to the final out_shardings that's used for lowering and compilation. JAX Array API #2472: Pull request #24387 synchronize by copybara-service bot
October 18, 2024 17:28 58s test_687174276
test_687174276
October 18, 2024 17:28 58s
[sharding_in_types] If out_aval.sharding is not None and the user spe… JAX Array API #2471: Commit 2153de4 pushed by copybara-service bot
October 18, 2024 17:28 59s main
main
October 18, 2024 17:28 59s
Push on main CodeQL #19726: by copybara-service bot
October 18, 2024 17:28 4m 41s
October 18, 2024 17:28 4m 41s
[sharding_in_types] If out_aval.sharding is not None and the user spe… CI #64222: Commit 2153de4 pushed by copybara-service bot
October 18, 2024 17:28 4m 2s main
main
October 18, 2024 17:28 4m 2s
PR #24387 CodeQL #19725: by copybara-service bot
October 18, 2024 17:27 4m 18s
October 18, 2024 17:27 4m 18s
[sharding_in_types] If out_aval.sharding is not None and the user specified out_sharding is None, concretize it with the device assignment available and add it to the final out_shardings that's used for lowering and compilation. CI #64221: Pull request #24387 synchronize by copybara-service bot
October 18, 2024 17:27 1m 23s test_687174276
test_687174276
October 18, 2024 17:27 1m 23s
[sharding_in_types] If out_aval.sharding is not None and the user specified out_sharding is None, concretize it with the device assignment available and add it to the final out_shardings that's used for lowering and compilation. JAX Array API #2470: Pull request #24387 synchronize by copybara-service bot
October 18, 2024 17:27 1m 0s test_687174276
test_687174276
October 18, 2024 17:27 1m 0s
Push on main CodeQL #19724: by copybara-service bot
October 18, 2024 17:22 4m 37s
October 18, 2024 17:22 4m 37s
[pallas] run_scoped now supports partial discharge. CI #64220: Commit f8a3c03 pushed by copybara-service bot
October 18, 2024 17:22 5m 45s main
main
October 18, 2024 17:22 5m 45s
[pallas] run_scoped now supports partial discharge. JAX Array API #2469: Commit f8a3c03 pushed by copybara-service bot
October 18, 2024 17:22 52s main
main
October 18, 2024 17:22 52s
PR #24382 CodeQL #19723: by copybara-service bot
October 18, 2024 17:09 4m 35s
October 18, 2024 17:09 4m 35s
[Pallas:MGPU] Fix the implementation of WGMMA with transposed RHS JAX Array API #2468: Pull request #24382 synchronize by copybara-service bot
October 18, 2024 17:09 1m 1s test_687261078
test_687261078
October 18, 2024 17:09 1m 1s
[Pallas:MGPU] Fix the implementation of WGMMA with transposed RHS CI #64219: Pull request #24382 synchronize by copybara-service bot
October 18, 2024 17:09 9m 26s test_687261078
test_687261078
October 18, 2024 17:09 9m 26s