PyTorch
API: torch.nn.functional.log_softmax
Strategy: Direct Mapping
JAX (Core)
API: jax.nn.log_softmax
Strategy: Direct Mapping
Keras
API: keras.activations.log_softmax
Strategy: Direct Mapping
TensorFlow
API: tf.math.log_softmax
Strategy: Direct Mapping
Apple MLX
API: mlx.nn.log_softmax
Strategy: Direct Mapping
Flax NNX
API: flax.nnx.log_softmax
Strategy: Direct Mapping