Add a customizable conv_transpose.#4938

copybara-service[bot] wants to merge 1 commit intomaingoogle/flax:mainfrom

test_782979395google/flax:test_782979395Copy head branch name to clipboard