아마존, PyTorch 코드로 Trainium2·3 시험하는 도구 공개

TokenPost · 2026. 10. 24. 오후 3:30:36

아마존($AMZN)이 기존 PyTorch 코드를 자체 AI 칩 Trainium2·3에서 실행하도록 돕는 도구 ‘TorchNeuron’의 공개 베타를 시작했다. 개발자는 코드에서 장치 설정을 바꿔 Trainium을 시험할 수 있다. AWS는 20일 공개 베타를 발표했다. PyTorch는 하루 앞서 TorchNeuron의 작동 방식을 소개했다. 이 도구는 Apache 2.0 라이선스의 오픈소스로 제공된다. PyTorch는 AI 모델을 만들고 학습하는 데 쓰이는 오픈소스 프레임워크다. 다른 칩으로 작업을 옮길 때는 하드웨어에 맞춰 코드를 손보고 결과를 검증해야 할 수 있다. TorchNeuron은 이 과정의 부담을 줄이는 데 초점을 맞췄다. 기존 코드에서 모델을 실행할 장치를 .to(neuron)으로... 더보기