code: def test_jax_device_put(
      *,
      dtype_and_x,
      test_flags,
      fn_tree,
      frontend,
      backend_fw,
      on_device,
  ):
      with BackendHandler.update_backend(backend_fw) as ivy_backend:
          dtype, x = dtype_and_x
          dtype = dtype[0]
          x = x[0]
  
          x = ivy_backend.asarray(x)
          if test_flags.as_variable and ivy_backend.is_float_dtype(
              dtype
          ):
              x = ivy_backend.functional.ivy.gradients._variable(x)
  
          device = ivy_backend.dev(x)
          x_on_dev = ivy_backend.functional.frontends.jax.device_put(
              x, on_device
          ).ivy_array
          dev_from_new_x = ivy_backend.dev(x_on_dev)
  
          # value test
          assert dev_from_new_x == device
  
file_path: ivy/ivy_tests/test_ivy/test_frontends/test_jax/test_general_functions.py
imports: import ivy
  import ivy_tests.test_ivy.helpers
  from ivy_tests.test_ivy.helpers import handle_frontend_test, BackendHandler
  from ivy.functional.frontends.jax import vmap
  from hypothesis import strategies
  import jax
label_desc: Use image node_id 5 for calling a global function with example usage: test_jax_device_put() without return types