flax
kernel_size parameter can be an int for 1d Conv and ConvTranspose
#551
Merged

Loading