__init__.py 164 B

123456
  1. # Copyright (c) Microsoft Corporation.
  2. # SPDX-License-Identifier: Apache-2.0
  3. # DeepSpeed Team
  4. from ..runtime.pipe import PipelineModule, LayerSpec, TiedLayerSpec