Skip to content
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.

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

jnp.einsum: add preferred_element_type argument #15479

Merged
merged 1 commit into from
Apr 10, 2023

Conversation

jakevdp
Copy link
Collaborator

@jakevdp jakevdp commented Apr 8, 2023

Fixes #15471

@jakevdp jakevdp self-assigned this Apr 8, 2023
@jakevdp jakevdp force-pushed the einsum-element-type branch from e64c89b to 3eb61e1 Compare April 8, 2023 16:25
@jakevdp jakevdp requested a review from yashk2810 April 10, 2023 17:49
@google-ml-butler google-ml-butler bot added kokoro:force-run pull ready Ready for copybara import and testing labels Apr 10, 2023
@copybara-service copybara-service bot merged commit 646339e into jax-ml:main Apr 10, 2023
@jakevdp jakevdp deleted the einsum-element-type branch April 10, 2023 18:07
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
pull ready Ready for copybara import and testing
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Add preferred_element_type to jax.numpy.einsum
3 participants