code: def test_jax_softmax(
      *,
      dtype_x_axis,
      test_flags,
      on_device,
      fn_tree,
      frontend,
      backend_fw,
  ):
      x_dtype, x, axis = dtype_x_axis
  
      helpers.test_frontend_function(
          input_dtypes=x_dtype,
          frontend=frontend,
          backend_to_test=backend_fw,
          test_flags=test_flags,
          fn_tree=fn_tree,
          on_device=on_device,
          rtol=1e-02,
          atol=1e-02,
          x=x[0],
          axis=axis,
      )
  
imports: from hypothesis import strategies
  import ivy_tests.test_ivy.helpers
  from ivy_tests.test_ivy.helpers import handle_frontend_test
label_desc: Use image node_id 22 for calling a global function with example usage: test_jax_softmax() without return types