in2_features

Auto-generated from flax_nnx_code_defs

PyTorchApple MLXFlax NNX

PyTorch

API: torch.nn.modules.linear.Bilinear.in2_features

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.nn.layers.normalization.BatchNorm.num_features

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: nnx.nn.attention.MultiHeadAttention.in_features

Strategy: Direct Mapping

Official Docs ↗