jax

c8ea86c9 - remove inlined jax.nn.initializers definitions, resolving TODO of levskaya et al

remove inlined jax.nn.initializers definitions, resolving TODO of levskaya et al fixes breakage from cl/655766534 aka https://github.com/google/jax/pull/21069 PiperOrigin-RevId: 655806010