code: def hinge_embedding_loss(
      input: JaxArray,
      target: JaxArray,
      *,
      margin: float = 1.0,
      reduction: str = "mean",
  ) -> JaxArray:
      zero_ = jnp.zeros([1], dtype=input.dtype)
  
      relu_part = jnp.maximum(margin - input, 0)
  
      loss = jnp.where(target == 1.0, input, zero_) + jnp.where(
          target == -1.0, relu_part, zero_
      )
      return _apply_loss_reduction(loss, reduction)
  
file_path: ivy/ivy/functional/backends/jax/experimental/losses.py
imports: import jax.numpy
  from typing import Optional
  from ivy.functional.backends.jax import JaxArray
  from ivy.func_wrapper import with_supported_device_and_dtypes
  from .None import backend_version
label_desc: Use image node_id 7 for calling a global function with example usage: hinge_embedding_loss(input, target) and returns: _apply_loss_reduction