flax
add truncated_normal to initializers
#3401
Merged

Loading