New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

in jax.lax.switch docstring, explain that all branch output shapes must match #25140

Open

mattjj opened this issue Nov 27, 2024 · 0 comments

Labels

better_errors documentation

Collaborator

mattjj commented Nov 27, 2024 •

edited

Loading

It's a bit hard to pick up from reading https://jax.readthedocs.io/en/latest/_autosummary/jax.lax.switch.html.

also we should improve the error message

mattjj added documentation better_errors labels

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment