    def model(self) -> Union[Tensor, Tuple[Tensor]]:
        """
        This function defines the AIT program.
        Returns an output tensor, or a tuple of output tensors.
        """
        pass
This function defines the AIT program.
Returns an output tensor, or a tuple of output tensors.
python/aitemplate/utils/serialization/ait_program.py