Skip to content

Actions: jax-ml/jax

Actions

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

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

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Make sure reshard and mesh_cast behave properly under eager mode CI #70505: Pull request #25996 synchronize by copybara-service bot
January 21, 2025 04:19 In progress test_717702155
test_717702155
January 21, 2025 04:19 In progress
Make sure reshard and mesh_cast behave properly under eager mode CI #70504: Pull request #25996 synchronize by copybara-service bot
January 21, 2025 04:04 11m 44s test_717702155
test_717702155
January 21, 2025 04:04 11m 44s
Make sure reshard and mesh_cast behave properly under eager mode CI #70503: Pull request #25996 opened by copybara-service bot
January 21, 2025 04:03 1m 12s test_717702155
test_717702155
January 21, 2025 04:03 1m 12s
[sharding_in_types] Move the calculation of new_mesh inside the decorator so that functions can be decorated with hidden_axes and visible_axes at the top level CI #70502: Pull request #25995 synchronize by copybara-service bot
January 21, 2025 03:41 11m 45s test_717691707
test_717691707
January 21, 2025 03:41 11m 45s
[sharding_in_types] Move the calculation of new_mesh inside the decor… CI #70501: Commit a943ebf pushed by copybara-service bot
January 21, 2025 03:41 11m 15s main
main
January 21, 2025 03:41 11m 15s
[sharding_in_types] Move the calculation of new_mesh inside the decorator so that functions can be decorated with hidden_axes and visible_axes at the top level CI #70500: Pull request #25995 synchronize by copybara-service bot
January 21, 2025 03:26 12m 6s test_717691707
test_717691707
January 21, 2025 03:26 12m 6s
[sharding_in_types] Move the calculation of new_mesh inside the decorator so that functions can be decorated with hidden_axes and visible_axes at the top level CI #70499: Pull request #25995 opened by copybara-service bot
January 21, 2025 03:15 11m 15s test_717691707
test_717691707
January 21, 2025 03:15 11m 15s
Fix attention backwards pass CI #70498: Pull request #25647 synchronize by Rifur13
January 21, 2025 02:51 12m 18s Rifur13:bwd_pass
Rifur13:bwd_pass
January 21, 2025 02:51 12m 18s
Clarify type hints for wrapped functions CI #70497: Pull request #25994 opened by ncoghlan
January 21, 2025 01:45 Action required ncoghlan:patch-1
ncoghlan:patch-1
January 21, 2025 01:45 Action required
Don't allow users to query tracer.sharding even under sharding in types mode. CI #70496: Pull request #25983 synchronize by copybara-service bot
January 20, 2025 23:12 11m 49s test_716442001
test_716442001
January 20, 2025 23:12 11m 49s
Don't allow users to query tracer.sharding even under sharding in t… CI #70495: Commit d50d1e2 pushed by copybara-service bot
January 20, 2025 23:12 11m 29s main
main
January 20, 2025 23:12 11m 29s
Don't allow users to query tracer.sharding even under sharding in types mode. CI #70494: Pull request #25983 synchronize by copybara-service bot
January 20, 2025 22:22 11m 3s test_716442001
test_716442001
January 20, 2025 22:22 11m 3s
Merge pull request #25984 from Rifur13:use_exp2 CI #70493: Commit 7f19b34 pushed by copybara-service bot
January 20, 2025 22:20 11m 54s main
main
January 20, 2025 22:20 11m 54s
Add test case for jit compile of vmap on gpu CI #70492: Pull request #25980 synchronize by codinglover222
January 20, 2025 21:56 Action required codinglover222:jit-vmap-compile-test
codinglover222:jit-vmap-compile-test
January 20, 2025 21:56 Action required
Add test case for jit compile of vmap on gpu CI #70491: Pull request #25980 synchronize by codinglover222
January 20, 2025 21:43 Action required codinglover222:jit-vmap-compile-test
codinglover222:jit-vmap-compile-test
January 20, 2025 21:43 Action required
Add test case for jit compile of vmap on gpu CI #70490: Pull request #25980 synchronize by codinglover222
January 20, 2025 21:23 Action required codinglover222:jit-vmap-compile-test
codinglover222:jit-vmap-compile-test
January 20, 2025 21:23 Action required
[better_errors] Ensure debug_info.arg_names is never None. CI #70489: Pull request #25992 synchronize by gnecula
January 20, 2025 20:32 8m 37s gnecula:debug_info_arg_names
gnecula:debug_info_arg_names
January 20, 2025 20:32 8m 37s
Don't allow users to query tracer.sharding even under sharding in types mode. CI #70488: Pull request #25983 synchronize by copybara-service bot
January 20, 2025 20:32 12m 2s test_716442001
test_716442001
January 20, 2025 20:32 12m 2s
[better_errors] Ensure debug_info.arg_names is never None. CI #70487: Pull request #25992 synchronize by gnecula
January 20, 2025 20:14 8m 40s gnecula:debug_info_arg_names
gnecula:debug_info_arg_names
January 20, 2025 20:14 8m 40s
[better_errors] Ensure debug_info.arg_names is never None. CI #70486: Pull request #25992 synchronize by gnecula
January 20, 2025 20:08 6m 18s gnecula:debug_info_arg_names
gnecula:debug_info_arg_names
January 20, 2025 20:08 6m 18s
Don't allow users to query tracer.sharding even under sharding in types mode. CI #70485: Pull request #25983 synchronize by copybara-service bot
January 20, 2025 20:07 11m 7s test_716442001
test_716442001
January 20, 2025 20:07 11m 7s
[better_errors] Ensure debug_info.arg_names is never None. CI #70484: Pull request #25992 synchronize by gnecula
January 20, 2025 20:00 8m 40s gnecula:debug_info_arg_names
gnecula:debug_info_arg_names
January 20, 2025 20:00 8m 40s
Merge pull request #25988 from gnecula:debug_info_tests CI #70483: Commit e41f4ca pushed by copybara-service bot
January 20, 2025 19:57 11m 42s main
main
January 20, 2025 19:57 11m 42s
[better_errors] Ensure debug_info.arg_names is never None. CI #70482: Pull request #25992 synchronize by gnecula
January 20, 2025 19:47 7m 22s gnecula:debug_info_arg_names
gnecula:debug_info_arg_names
January 20, 2025 19:47 7m 22s
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 #70481: Pull request #25978 synchronize by copybara-service bot
January 20, 2025 19:39 13m 38s test_716879595
test_716879595
January 20, 2025 19:39 13m 38s