num_heads

Auto-generated from flax_nnx_code_defs

PyTorchApple MLXFlax NNX

PyTorch

API: torch.nn.modules.activation.MultiheadAttention.num_heads

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.nn.layers.transformer.MultiHeadAttention.num_heads

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: nnx.nn.attention.MultiHeadAttention.num_heads

Strategy: Direct Mapping

Official Docs ↗