code: def _pad_in_dim(x, low=0, high=0, interior=0, fill_value=0, axis=0):
      pads = [(0, 0, 0)] * x.ndim
      pads[axis] = (low, high, interior)
      return lax.pad(x, jnp.array(fill_value, x.dtype), pads)
  
file_path: jax/jax/_src/lax/qdwh.py
imports: from __future__ import annotations
  import functools
  import jax
  import jax.numpy
  from jax import lax
  from jax._src import core
  from jax._src.lax import linalg
label_desc: Use image node_id 2 for calling a global function with example usage: _pad_in_dim(x, low, high, interior, fill_value, axis) and returns: lax