from .amp import init, half_function, float_function, promote_function,\
    register_half_function, register_float_function, register_promote_function,\
    register
from .multi_tensor_apply import MultiTensorApply
