flax

a6089931 - Added glorot and he initializers