from .device import (TorchDeviceInfo, TorchDevicesInfo, get_available_devices, get_cpu_device)