class_docstr: Tests for computing the maximum entropy of a given state.
class_docstr_tok: ["Tests","for","computing","the","maximum","entropy","of","a","given","state","."]
docstr: Test `max_entropy` differentiability with jax.
docstr_tok: ["Test","`","max_entropy","`","differentiability","with","jax","."]
code: def test_max_entropy_grad_jax(
      self, params, wires, base, check_state, jit
  ):
      """Test `max_entropy` differentiability with jax."""
      jnp = jax.numpy
      jax.config.update("jax_enable_x64", True)
  
      params = jnp.array(params)
  
      max_entropy_grad = jax.grad(qml.math.max_entropy)
  
      if jit:
          gradient = jax.jit(
              max_entropy_grad, static_argnums=[1, 2, 3]
          )(params, tuple(wires), base, check_state)
      else:
          gradient = max_entropy_grad(params, wires, base, check_state)
  
      assert qml.math.allclose(gradient, 0.0)
  
imports: import pytest
  import pennylane
  from pennylane import numpy
label_desc: Use image node_id 5 for calling the TestMaxEntropy obj's underlying member method code with example usage: obj.test_max_entropy_grad_jax(params, wires, base, check_state, jit) without return types