Introducing TorchNeuron: Native PyTorch backend for AWS Trainium

PyTorch introduces TorchNeuron, a native backend for AWS Trainium, targeting complex models like dense transformers and diffusion models. This development aims to enhance AI research and model deployment on PyTorch. TorchNeuron is designed to support various model architectures, including mixture-of-experts transformers and state-space models.