code: def funnel_log_density(params):
      return norm.logpdf(
          params[0], 0, jnp.exp(params[1])
      ) + norm.logpdf(params[1], 0, 1.35)
  
file_path: jax/examples/advi.py
imports: from functools import partial
  import matplotlib.pyplot
  from jax import jit, grad, vmap
  from jax import random
  from jax.example_libraries import optimizers
  import jax.numpy
  import jax.scipy.stats.norm
label_desc: Use image node_id 7 for calling a global function with example usage: funnel_log_density(params) and returns: unknown