Files
DeepSpeed/deepspeed/ops/adam/__init__.py

3 lines
73 B
Python
Executable File

from .cpu_adam import DeepSpeedCPUAdam
from .fused_adam import FusedAdam