Fix Neuron engine for SDK 2.28+ and add Trainium (trn2) support#659

Open

jimburtoft wants to merge 5 commits intomichaelfeil:mainfrom

jimburtoft:fix/neuron-sdk-2.28-compat

Commits on Feb 28, 2026

Fix Neuron engine compatibility with optimum >= 2.0 and Neuron SDK 2.28

jimburtoft
authored and
Ubuntu
committed

Commits on Mar 1, 2026

Add CHECK_OPTIMUM guard and cross-instance benchmarks

jimburtoft
authored and
Ubuntu
committed

Commits on Mar 2, 2026

Use data parallelism instead of tensor parallelism for Neuron

jimburtoft
authored and
Ubuntu
committed
Add trn2.3xlarge benchmarks and data parallelism examples to Neuron docs
Ubuntu
committed
Address code review feedback: fix BetterTransformer guard, add NEURON_NUM_CORES env var
Ubuntu
committed