AWS, TorchNeuron 발표: AWS Trainium용 네이티브 PyTorch 지원(공개 베타)

게시된 날짜: 2026년 10월 20일

AWS는 오늘 AWS Trainium용 오픈 소스 네이티브 PyTorch 백엔드인 TorchNeuron의 공개 베타를 출시했습니다. Trainium은 딥 러닝 AI 훈련과 추론 워크로드에 뛰어난 성능과 비용 효율성을 제공하도록 설계된 AI 칩 포트폴리오입니다. TorchNeuron을 사용하면 AI 과학자와 연구원이 Trainium에서 기본적으로 PyTorch를 사용하여 모델을 실험하고 구축할 수 있습니다.

TorchNeuron을 통해 개발자는 .to ('neuron')로 전환하기만 하면 Trainium에서 PyTorch 코드를 활성화하는 동시에 즉시 실행, torch.compile, 표준 PyTorch API로 완벽한 PyTorch 호환성을 유지할 수 있습니다. 또한 TorchNeuron은 Neuron Kernel Interface(NKI)와 통합되므로 Trainium 하드웨어에 대한 사용자 지정 연산자 및 성능 최적화가 가능합니다.

TorchNeuron은 Amazon EC2 Trainium2 인스턴스 및 Amazon EC2 Trainium3 인스턴스용 TorchNeuron github 리포지토리를 통해 오픈 소스로 사용할 수 있습니다.

TorchNeuron과 AWS Trainium 액셀러레이터에 대해 자세히 알아보려면 AWS Neuron 페이지를 참조하세요.