code: def log_softmax(
      x: JaxArray,
      /,
      *,
      axis: Optional[int] = -1,
      complex_mode: Literal["split", "magnitude", "jax"] = "jax",
      out: Optional[JaxArray] = None,
  ):
      return jax.nn.log_softmax(x, axis)
  
file_path: ivy/ivy/functional/backends/jax/activations.py
imports: import jax
  import jax.numpy
  from typing import Optional, Union, Literal
  from ivy.functional.backends.jax import JaxArray
label_desc: Use image node_id 8 for calling a global function with example usage: log_softmax() and returns: jax