xavier_normalΒΆ Auto-generated from flax_nnx_code_defs PyTorchFlax NNX PyTorchAPI: torch.nn.init.xavier_normalStrategy: Direct MappingOfficial Docs βFlax NNXAPI: nnx.rnglib.Rngs.xavier_normalStrategy: Direct MappingOfficial Docs β