code: def _randn(prototype, shape, rng_key=None):
      assert isinstance(shape, tuple)
      return jax.random.normal(rng_key, shape, dtype=prototype.dtype)
  
imports: import numbers
  import typing
  import jax.numpy
  import jax.random
  import numpy
  from jax import lax
  from jax.core import Tracer
  from jax.scipy.linalg import cho_solve, solve_triangular
  from jax.scipy.special import expit, gammaln, logsumexp
  from ..None import ops
label_desc: Use image node_id 36 for calling a global function with example usage: _randn(prototype, shape, rng_key) and returns: jax