__init__.py 427 B

12345678910111213
  1. '''Copyright The Microsoft DeepSpeed Team'''
  2. from . import adam
  3. from . import adagrad
  4. from . import lamb
  5. #from ..git_version_info_installed import installed_ops as __installed_ops__
  6. #if __installed_ops__['sparse_attn']:
  7. from . import sparse_attention
  8. from . import transformer
  9. from .transformer import DeepSpeedTransformerLayer, DeepSpeedTransformerConfig
  10. from ..git_version_info import compatible_ops as __compatible_ops__