code: def init_params(rng):
      mlp_variables = model.init({"params": rng}, X)
      return mlp_variables["params"]
  
imports: import jax
  from jax import numpy, jit
  from dense import Dense
label_desc: Use image node_id 3 for calling a global function with example usage: init_params(rng) and returns: mlp_variables