jax.nn.initializers.glorot_uniform

jax.nn.initializers.glorot_uniform(in_axis=-2, out_axis=-1, dtype=<class 'jax._src.numpy.lax_numpy.float32'>)