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.
DeviceMPS	Device to support MPS, for training on Apple's M-series chips.
DeviceTPU	An extension of Device for TPUs.