    def __init__(
        self,
        min_value: IntVar,
        max_value: IntVar,
    ):
        """Initializes a JaggedDim.

        Parameters
        ----------
        min_value : IntVar
            Minimum possible value of the jagged dimension.
        max_value : IntVar
            Maximum possible value of the jagged dimension.
        """
        if isinstance(min_value, int):
            min_value = IntImm(min_value)
        if isinstance(max_value, int):
            max_value = IntImm(max_value)

        if min_value.lower_bound() < 0:
            raise ValueError(f"{min_value=}, but must be non-negative.")
        if min_value.lower_bound() > max_value.upper_bound():
            raise ValueError(f"{min_value=} can't be larger than {max_value=}.")

        super().__init__()

        self._attrs["values"] = [min_value, max_value]
        self._attrs["offsets"] = None
Initializes a JaggedDim.

Parameters
----------
min_value : IntVar
Minimum possible value of the jagged dimension.
max_value : IntVar
Maximum possible value of the jagged dimension.
python/aitemplate/compiler/base.py