code: def reparameterize(rng, mean, logvar):
      std = jnp.exp(0.5 * logvar)
      eps = random.normal(rng, logvar.shape)
      return mean + eps * std
  
file_path: flax/examples/vae/models.py
imports: from flax import linen
  from jax import random
  import jax.numpy
label_desc: Use image node_id 1 for calling a global function with example usage: reparameterize(rng, mean, logvar) and returns: unknown