Skip to content

Conversation

@samanklesaria
Copy link
Collaborator

@samanklesaria samanklesaria commented Sep 29, 2025

Closes #4593 by documenting the highly confusing padding argument for ConvTranspose.

Note that my removal of the duplicate nnx import line in linear.py was necessary to satisfy the pre-commit checks.

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.

Flax nnx ConvTranspose Does Not Restore Input Shape When Used with Conv (Unexpected Behavior)

1 participant