class_docstr: Domain class describing a rectangular domain
  
  Attributes:
      size (Tuple[int]): The size of the domain in absolute units.
      dx (Tuple(float)): The unit of measure
class_docstr_tok: ["Domain","class","describing","a","rectangular","domain","Attributes",":","size","(","Tuple","[","int","]",")",":","The","size","of","the","domain","in","absolute","units",".","dx","(","Tuple","(","float",")",")",":","The","unit","of","measure"]
code: def _make_grid_from_axis(axis):
      return jnp.stack(jnp.meshgrid(*axis, indexing="ij"), axis=-1)
  
file_path: jaxdf/jaxdf/geometry.py
imports: from functools import reduce
  from typing import Iterable
  import equinox
  import jax
  from jax import numpy
  from jax import random
  from .mods import Module
label_desc: Use image node_id 8 for calling the Domain obj's underlying member method code with example usage: obj._make_grid_from_axis(axis) and returns: jnp