flax
replaced zeros and ones initializers with zeros_init and ones_init builder functions
#2807
Merged
Merged