kernel_sizeΒΆ Auto-generated from flax_nnx_code_defs PyTorchFlax NNX PyTorchAPI: torch.nn.modules.fold.Unfold.kernel_sizeStrategy: Direct MappingOfficial Docs βFlax NNXAPI: nnx.nn.linear.ConvTranspose.kernel_sizeStrategy: Direct MappingOfficial Docs β