Oossa

i14, AWS Trainium2에서 Laplace 3B 학습 속도 초당 12만 토큰

3B 파라미터 선형 어텐션 모델은 trn2.48xlarge 인스턴스에서 초당 120,000토큰으로 학습됐으며, 초당 11,544토큰을 처리했습니다.

짧은 소식작성: Oossa 게시일: 1 분 읽기

i14는 AWS Trainium2와 TorchNeuron Native를 사용해 선형 어텐션 모델 Laplace 3B를 학습하고 서빙했습니다. 단일 trn2.48xlarge 인스턴스에서 모델은 초당 약 120,000토큰으로 학습됐고, 동시에 실행되는 시퀀스 1,280개에 걸쳐 초당 11,544개의 출력 토큰을 생성할 수 있었습니다. 이 아키텍처는 컨텍스트 길이와 관계없이 시퀀스당 추론 상태를 29.5MB로 일정하게 유지합니다.

왜 중요한가

메모리 사용량을 예측할 수 있어 장시간 실행되는 AI 에이전트가 비용 증가 없이 훨씬 더 많은 텍스트를 처리할 수 있습니다.

이 기사가 도움이 되었나요?
공유

다음 읽을거리

Oossa · 뉴스레터

이번 주 AI, 쉽게 정리

매주 월요일, 알아둘 만한 소식을 쉬운 말로. 무료, 스팸 없음.