code: def digamma(x):
      return scipy.special.digamma(x).astype(x.dtype)
  
imports: import builtins
  import collections
  import itertools
  import numpy
  import opt_einsum
  import scipy.special
  from jax._src import dtypes
  from jax._src import util
label_desc: Use image node_id 5 for calling a global function with example usage: digamma(x) and returns: scipy