[Neuron] Add AWS Neuron (Trainium/Inferentia) as an officially supported device (#13289)
* draft:add neuron as a legit backend
* feat: neuron-specific changes in the pipeline
* tests: eager tests
* fix: style
* fix:apr_02 beta
* cleanup: remove tp part, for another pr
* fix: restore ring_anything to ContextParallelConfig after over-aggressive TP cleanup
The previous cleanup commit removed TensorParallelConfig (correct) but also
accidentally removed ring_anything from ContextParallelConfig (incorrect).
ring_anything is a context-parallel feature referenced in context_parallel.py
and must remain in the config.
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
* removal: style fix
* tests:sdxl + flux2
* tests: simplify
* tests: simplify
* fix style
* fix style
* fix style for doc-builder
* review: address comments
* review:apply suggestion for the fix of index_for_timtestep
* review: stronger guard on image slice
* Apply suggestions from code review
Co-authored-by: YiYi Xu <yixu310@gmail.com>
* fix: when set_begin_index not implemented for scheduler
* review:add maybe_adjust_dtype_for_device and apply to all models with downcasting needs
* fix: dependency
* Update src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py
Co-authored-by: YiYi Xu <yixu310@gmail.com>
* review: apply maybe_adjust_dtype_for_device in pixart pipe
* review: update .ai/models.md
---------
Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>
Co-authored-by: YiYi Xu <yixu310@gmail.com>