env/Lib/site-packages/functorch/experimental/control_flow.py

from torch._higher_order_ops.cond import (  # noqa: F401
    cond,
    UnsupportedAliasMutationException,
)

from ._map import map  # noqa: F401