__init__.py 119 B

1234
  1. '''Copyright The Microsoft DeepSpeed Team'''
  2. from .cpu_adam import DeepSpeedCPUAdam
  3. from .fused_adam import FusedAdam