devices

Modules

composer.trainer.devices.device	The base Device class.
composer.trainer.devices.device_cpu	The CPU device used for training.
composer.trainer.devices.device_gpu	The GPU device used for training.
composer.trainer.devices.device_hparams_registry	The Hparams used to construct devices.
composer.trainer.devices.device_mps	The Apple M-series device used for training.
composer.trainer.devices.device_tpu	The TPU device used for training.

Module for devices on which models run.

Used by Trainer in order to train on different devices such as GPU and CPU.

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.