deep_ep/__init__.py (4 lines of code) (raw):

import torch from .utils import EventOverlap from .buffer import Buffer # noinspection PyUnresolvedReferences from deep_ep_cpp import Config