flax

b5ec12b8 - expose ones and zeros initializers