__init__.py 496 B

1234567891011121314151617
  1. # Copyright (c) Microsoft Corporation.
  2. # SPDX-License-Identifier: Apache-2.0
  3. # DeepSpeed Team
  4. from . import adam
  5. from . import adagrad
  6. from . import lamb
  7. from . import lion
  8. #from ..git_version_info_installed import installed_ops as __installed_ops__
  9. #if __installed_ops__['sparse_attn']:
  10. from . import sparse_attention
  11. from . import transformer
  12. from .transformer import DeepSpeedTransformerLayer, DeepSpeedTransformerConfig
  13. from ..git_version_info import compatible_ops as __compatible_ops__