Skip to content

Conversation

@copybara-service
Copy link

Use .input_formats and .output_formats in place of .input_layouts and .output_layouts respectively.

JAX is undergoing a rename of the contents of jax.experimental.layouts in preparation for its graduation from experimental:

"Formats" are replacing "layouts", and specifically Layout -> Format
"Layouts" are replacing "device local layouts", and specifically DeviceLocalLayout -> Layout
This is an incremental update carrying out #1.

@copybara-service copybara-service bot force-pushed the test_773109310 branch 2 times, most recently from 8404b16 to ed13529 Compare June 19, 2025 01:37
…s` and `.output_layouts` respectively.

JAX is undergoing a rename of the contents of jax.experimental.layouts in preparation for its graduation from experimental:

"Formats" are replacing "layouts", and specifically Layout -> Format
"Layouts" are replacing "device local layouts", and specifically DeviceLocalLayout -> Layout
This is an incremental update carrying out #1.

PiperOrigin-RevId: 773141876
@copybara-service copybara-service bot merged commit dcd2f5e into main Jun 19, 2025
@copybara-service copybara-service bot deleted the test_773109310 branch June 19, 2025 01:39
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants