Is there a constant initializer for params in Flax/JAX?

[marcvanzee]

(Original question by @yangli169)

[marcvanzee]

Answer by @jacobaustin123: I don't think so, but you could make your own with:

def constant(key, shape, dtype=jnp.float32, value=0.0): 
  return jnp.full(shape, value, dtype)