AWS는 파이토치 기반 AI 모델을 자사 AI 칩 트레이니엄에서 거의 수정 없이 실행할 수 있게 하는 소프트웨어 토치뉴런을 공개한다. 이 소프트웨어는 파이토치가 새 하드웨어를 연결할 때 쓰는 표준 메커니즘을 활용해 트레이니엄을 기존 개발 환경에 붙인다. 개발자는 코드를 처음부터 다시 쓰는 대신 칩 지정 같은 설정을 바꿔 적용할 수 있고, 기존 테스트·디버깅·모델 확장 도구도 큰 변경 없이 이어서 쓸 수 있다. 토치뉴런은 아파치 2.0 라이선스로 제공되며 트레이니엄2와 트레이니엄3를 지원한다.
이 기술의 의미는 칩 성능 비교를 넘어, 소프트웨어 이식과 검증에 드는 시간을 줄인다는 점에 있다. 아마존은 동일한 학습 스크립트를 기존 칩과 트레이니엄에서 각각 실행하는 시연에서 칩 설정만 다르고 모델은 같은 내용을 같은 속도로 학습했다고 설명했다. 실제 성능과 비용은 모델 구조, 작업 종류, 하드웨어 구성, 최적화 수준에 따라 달라질 수 있다. 그래도 기존 파이토치 코드를 유지한 채 다른 하드웨어를 시험할 수 있으면, 개발자는 특정 칩에 덜 묶이고 인프라 활용 방식도 더 유연해진다.