Skip to content

[tensor-shapes] improve jax.numpy.fft stubs - #4993

Closed
jakevdp wants to merge 3 commits into
facebook:mainfrom
jakevdp:jax-fft-shape
Closed

jakevdp wants to merge 3 commits into
facebook:mainfrom
jakevdp:jax-fft-shape

Conversation

@jakevdp

@jakevdp jakevdp commented Sep 20, 2026

Copy link
Copy Markdown
Contributor

Propagate shapes in more cases & avoid overloads in python stubs.

@meta-cla meta-cla Bot added the cla signed label Sep 20, 2026
@github-actions github-actions Bot added google issues from google size/xl labels Sep 20, 2026
@meta-codesync

meta-codesync Bot commented Sep 20, 2026

Copy link
Copy Markdown
Contributor

This pull request has been imported. If you are a Meta employee, you can view this in D120930728. (Because this pull request was imported automatically, there will not be any future comments.)

Comment thread tensor-shapes/pyrefly-jax-stubs/jax-stubs/_shapes.pyi

@stroxler stroxler left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Back to you for the question about shared helpers. If you'd rather use helpers that's fine, but my first instinct is that it would make the _shapes code easier to maintain

@stroxler stroxler left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Review automatically exported from Phabricator review in Meta.

@github-actions github-actions Bot added size/xl and removed size/xl labels Sep 21, 2026
@meta-codesync meta-codesync Bot closed this in b23f8e5 Sep 21, 2026
@meta-codesync meta-codesync Bot added the Merged label Sep 21, 2026
@meta-codesync

meta-codesync Bot commented Sep 21, 2026

Copy link
Copy Markdown
Contributor

This pull request has been merged in b23f8e5.

@jakevdp
jakevdp deleted the jax-fft-shape branch September 21, 2026 21:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants