GitHub Repo
AWS Publicly Tests TorchNeuron, Bringing Native PyTorch Execution to Trainium; Compilation Still Limited to Static Shapes
On October 20, AWS released the open beta of open-source TorchNeuron, supporting PyTorch eager execution and model compilation on Trainium2 and Trainium3. The source documentation also lists limitations for dynamic shapes, data types, and operator compatibility, so migrations still require item-by-item validation.