flax
updated `nnx.Conv` and `nnx.ConvTranspose`
#3974
Merged

Loading