DeviceNeuron#

class composer.devices.DeviceNeuron[source]#

An extension of Device for Neuron devices (Trn, Inf).

When running on Trn, we automatically set export PJRT_DEVICE=NEURON.