class_docstr: Tests that the template is compatible with all interfaces.
docstr: Tests jax tensors.
docstr_tok: ["Tests","jax","tensors","."]
code: def test_jax(self, tol, features, pad_with, normalize):
      """Tests jax tensors."""
      import jax.numpy as jnp
  
      features = jnp.array(features)
  
      dev = qml.device("default.qubit")
  
      circuit = qml.QNode(circuit_template, dev, interface="jax")
      circuit2 = qml.QNode(circuit_decomposed, dev)
  
      res = circuit(features, pad_with, normalize)
      res2 = circuit2(features, pad_with)
  
      assert qml.math.allclose(res, res2, atol=tol, rtol=0)
  
imports: import pytest
  import numpy
  import pennylane
  from pennylane import numpy
label_desc: Use image node_id 3 for calling the TestInterfaces obj's underlying member method code with example usage: obj.test_jax(tol, features, pad_with, normalize) without return types