composer.devices

Module for devices on which models run.

Classes

Device	Abstract class for a device on which a model runs.
DeviceCPU	An extension of Device for CPUs.
DeviceGPU	An extension of Device for GPUs.
DeviceHPU	An extension of Device for HPUs.
DeviceMPS	Device to support MPS, for training on Apple's M-series chips.
DeviceTPU	An extension of Device for TPUs.