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

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

jheek
jheek jheek requested a review from marcvanzee marcvanzee 3 years ago
marcvanzee
marcvanzee requested changes on 2022-01-25
jheek
jheek commented on 2022-01-25
jheek jheek requested a review from marcvanzee marcvanzee 3 years ago
PhilipVinc
PhilipVinc approved these changes on 2022-01-26
jheek Add support for non-float32 normalization for linen normalization lay…
c87b54b7
jheek jheek force pushed to c87b54b7 3 years ago
marcvanzee
marcvanzee approved these changes on 2022-01-26
jheek jheek added pull ready
copybara-service copybara-service merged ca683887 into main 3 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone