flax
Add support for non-float32 normalization for linen normalization layers
#1804
Merged

Commits
  • Add support for non-float32 normalization for linen normalization layers.
    jheek committed 3 years ago
Loading