class_docstr: Tests jit targeting to different backends.
class_docstr_tok: ["Tests","jit","targeting","to","different","backends","."]
code: def test_closed_over_values_device_placement(self):
      # see https://github.com/google/jax/issues/1431
      def f():
          return jnp.add(3.0, 4.0)
  
      self.assertNotEqual(
          jax.jit(f)().devices(), {jax.devices("cpu")[0]}
      )
      self.assertEqual(
          jax.jit(f, backend="cpu")().devices(), {jax.devices("cpu")[0]}
      )
  
file_path: jax/tests/multibackend_test.py
imports: from functools import partial
  from absl.testing import absltest
  import numpy
  import numpy.random
  from unittest import SkipTest
  import jax
  from jax._src import test_util
  from jax import numpy
  from jax import config
label_desc: Use image node_id 6 for calling the MultiBackendTest obj's underlying member method code with example usage: obj.test_closed_over_values_device_placement() and returns: jnp