PyTorch
API: torch.nn.modules.activation.Softmax
Strategy: Direct Mapping
Keras
API: keras.layers.Softmax
Strategy: Direct Mapping
TensorFlow
API: tf.math.softmax
Strategy: Direct Mapping
Apple MLX
API: mlx.nn.layers.activations.Softmax
Strategy: Direct Mapping
Flax NNX
API: nnx.softmax
Strategy: Direct Mapping