code: def new_linear_params(n_in, n_out):
      W = np.random.randn(n_out, n_in).astype(
          np.float32
      )  # TODO(beam2d): not supported in chx
      W /= np.sqrt(n_in)  # TODO(beam2d): not supported in chx
      W = chx.array(W)
      b = chx.zeros(n_out, dtype=chx.float32)
      return W, b
  
file_path: chainer/chainerx_cc/examples/mnist_py/train_mnist.py
imports: import argparse
  import gzip
  import pathlib
  import time
  import numpy
  import chainerx
label_desc: Use image node_id 1 for calling a global function with example usage: new_linear_params(n_in, n_out) and returns: W,  b