Adding a new transposed convolution function (similar to `torch.nn.ConvTranspose2d()`)
Open
Priority: P2 - no schedule
Status: blocked
- Dominant language
- Jupyter Notebook
- Stars
- 7.3k
- Forks
- 833
- Avg merge
- 5h 11m
- Merged PRs (30d)
- 5
Description
Adding a transposed convolution as proposed in
https://github.com/google/jax/pull/5772
would also be very useful when porting models from PyTorch to Flax (as in #1848).
Contributor guide
Assessment
This issue has not been assessed yet.