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