device_mps#

The Apple M-series device used for training.

Classes

DeviceMPS

Device to support MPS, for training on Apple's M-series chips.
class composer.trainer.devices.device_mps.DeviceMPS[source]#

Bases: composer.trainer.devices.device.Device

Device to support MPS, for training on Appleโ€™s M-series chips.

This class takes no arguments.