code: def sort_key_val(*args, **kwargs):
      """Sorts keys along dimension and applies same permutation to values."""
      return backend()["sort_key_val"](*args, **kwargs)
  
imports: import contextlib
  import enum
  import gin
  from trax.fastmath.jax import JAX_BACKEND
  from trax.fastmath.numpy import NUMPY_BACKEND
  from trax.fastmath.tf import TF_BACKEND
label_desc: Use image node_id 10 for calling a global function with example usage: sort_key_val() and returns: backend