class_docstr: Test the utility function detach.
class_docstr_tok: ["Test","the","utility","function","detach","."]
docstr: Test that detach works with JAX.
docstr_tok: ["Test","that","detach","works","with","JAX","."]
code: def test_jax(self, use_jit):
      """Test that detach works with JAX."""
  
      x = jax.numpy.array(0.3)
      func = (
          jax.jit(fn.detach, static_argnums=1) if use_jit else fn.detach
      )
      jac = jax.jacobian(func)(x)
      assert jax.numpy.isclose(jac, 0.0)
  
imports: import autoray
  import numpy
  import pytest
  import scipy
  from autoray import numpy
  from pennylane import math
  from pennylane import numpy
  from pennylane import grad
label_desc: Use image node_id 3 for calling the TestDetach obj's underlying member method code with example usage: obj.test_jax(use_jit) without return types