Skip to content

Add support for axis_name and axis_index_groups to lax.ragged_all_to_all #9419

Add support for axis_name and axis_index_groups to lax.ragged_all_to_all

Add support for axis_name and axis_index_groups to lax.ragged_all_to_all #9419

build (3.11)

succeeded Jan 29, 2025 in 2m 46s
Set up job
1s
Checkout jax
1s
Checkout array-api-tests
2s
Set up Python 3.11
0s
Install dependencies
16s
Run the test suite
2m 20s
Post Set up Python 3.11
0s
Post Checkout array-api-tests
0s
Post Checkout jax
0s
Complete job
0s