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