code: def move_axis(x, source, dest):
      """Move axis from source to dest."""
      return jnp.moveaxis(x, source, dest)
  
imports: import functools
  import jax
  from jax import lax
  from jax import random
  from jax import tree_util
  import jax.numpy
label_desc: Use image node_id 19 for calling a global function with example usage: move_axis(x, source, dest) and returns: jnp