flax
Remove the need for Conv and ConvTranspose to know the precise batch size.
#4877
Merged
Merged