flax
Add GroupNorm to NNX normalization layers
#4095
Merged

Loading