The Diffusers library now treats AWS Neuron chips (Trainium and Inferentia) as a first-class compute backend alongside CUDA, MPS, XPU, and MLU, adding device registration, backend dispatch, and pipeline methods for eager-mode inference on these accelerators.
Registers Neuron in all Diffusers backend dispatch tables and provides enable_neuron_compile() and neuron_warmup() pipeline methods for compiling and warming up models on Trainium/Inferentia hardware.
Users of the Diffusers library can now run popular image generation models such as SDXL and PixArt on AWS Trainium and Inferentia chips without patching internals, opening a lower-cost inference option outside GPU clouds. Operators choosing AWS inf2 or trn instances get official support and warmup utilities that front-load neuronx-cc compilation so production latency is predictable. The initial validation covers only three models under eager mode; torch.compile support, tensor parallelism, and video-model sequence parallelism remain on the roadmap, so coverage gaps and performance regressions on other architectures need continued monitoring.