code: def _is_simplex(x: Array) -> Array:
      x_sum = jnp.sum(x, axis=0)
      return jnp.all(x > 0, axis=0) & (abs(x_sum - 1) < 1e-6)
  
file_path: jax/jax/_src/scipy/stats/dirichlet.py
imports: import scipy.stats
  from jax import lax
  import jax.numpy
  from jax._src.lax.lax import _const
  from jax._src.numpy.util import promote_dtypes_inexact, _wraps
  from jax.scipy.special import gammaln, xlogy
  from jax._src.typing import Array, ArrayLike
label_desc: Use image node_id 1 for calling a global function with example usage: _is_simplex(x) and returns: unknown