log_softmax =========== Auto-generated from jax_code_defs .. raw:: html

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