flax
[linen] fix DenseGeneral init
#3834
Merged

Loading