code: def _make_randn_fn(n, k, m):
      rng = np.random.default_rng(1234)
      tall_skinny = rng.standard_normal((n, k))
  
      def randn_action(x):
          ts = jnp.array(tall_skinny, dtype=x.dtype)
          p = jax.lax.Precision.HIGHEST
          return ts.dot(ts.T.dot(x, precision=p), precision=p)
  
      _, s, _ = np.linalg.svd(tall_skinny, full_matrices=False)
      return randn_action, s**2, m
  
imports: import functools
  import re
  import os
  import unittest
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  import scipy.linalg
  import scipy.sparse
  import jax
  from jax import config
  from jax._src import test_util
  from jax.experimental.sparse import linalg, bcoo
  import jax.numpy
label_desc: Use image node_id 11 for calling a global function with example usage: _make_randn_fn(n, k, m) and returns: randn_action,  unknown,  m, ts