code: def nucleaus_sample(key, logits, _, top_p=0.9, temp=1, top_k=None):
      logits = nucleaus_filter(logits, top_p, top_k=top_k)
  
      return softmax_sample(key, logits, None, temp=temp)
  
imports: import jax
  import jax.numpy
label_desc: Use image node_id 3 for calling a global function with example usage: nucleaus_sample(key, logits, _, top_p, temp, top_k) and returns: softmax_sample