AWS, PyTorch 앱을 트레이니엄에서 실행하는 'TorchNeuron' 공개
AWS는 코드 수정 없이 설정 한 줄만 바꾸면 PyTorch로 작성된 AI 앱이 트레이니엄 칩에서 실행된다고 밝혔다.
AWS가 2026년 10월 19일 PyTorch로 작성된 AI 애플리케이션을 아마존의 트레이니엄(Trainium) 칩에서 실행할 수 있게 해주는 개발자 도구 'TorchNeuron'을 공개했다고 테크크런치가 보도했다. 설정 한 줄만 바꾸면 되는 방식이다.
아마존 측은 TorchNeuron이 코드 재작성을 요구하지 않으며, PyTorch에 익숙한 개발자라면 별도 학습 없이 바로 사용할 수 있다고 설명했다. 아마존은 AI 앱 가운데 약 85%가 PyTorch로 작성된다고 밝히고 있다.
AWS는 프레임워크를 특정 칩에 묶는 코드를 없애기 위해 관련 코드를 PyTorch 본 프로젝트에 업스트림으로 기여하고 있다고 전했다. 이는 PyTorch 애플리케이션이 특정 하드웨어에 종속되지 않도록 하려는 작업이다.
AWS 측은 트레이니엄이 학습과 추론을 한 칩에서 처리하며 자사 클라우드 인프라와 통합돼 있다고 설명했다. 이번 발표에 대해 AWS 앤나푸르나 랩스(AWS Annapurna Labs)의 기술 소프트웨어 부사장인 콜린 브레이스(Colin Brace)가 언급했다.
PyTorch와 칩 생태계
PyTorch는 2022년부터 오픈소스 프로젝트를 관리하는 리눅스 재단(Linux Foundation) 산하에 속해 있다. 원래는 페이스북이 개발했다. 엔비디아는 수년간 자사 CUDA 소프트웨어가 PyTorch로 작성된 AI 앱을 지원하도록 작업해 왔다.
한눈에 보기
TorchNeuron은 무엇인가?
PyTorch로 작성된 AI 앱을 설정 한 줄 변경만으로 아마존 트레이니엄 칩에서 실행할 수 있게 하는 AWS의 개발자 도구다.
코드를 다시 작성해야 하나?
아마존은 코드 재작성이 필요 없으며 PyTorch에 익숙한 개발자가 바로 사용할 수 있다고 밝혔다.