AI LAB / KEYWORD DICTIONARY

AIアクセラレーター

AWS Trainium(AWSトレニアム)とは?

大規模なAIモデルの学習と推論を高い性能と経済性で処理するために設計されたAWSの専用AIチップ。

仕組みと使いどころ

AIの学習と推論を対象とし、サーバー、ネットワーク、ソフトウェア、サービスと一体で利用する専用チップである。PyTorch、vLLM、Hugging Face、Rayなどを利用でき、SDKは通常の開発から命令セットに近いカーネル開発まで支援する。大規模構成への拡張も想定されているが、実際の性能や経済性はチップだけでなく、モデル、メモリー、通信、運用環境を含む構成に左右される。

AWS Trainiumの活用例

たとえば、こんな場面で

大規模モデルの学習基盤や推論基盤で、対応フレームワークから専用計算資源を利用する構成が考えられる。

使うときのポイント

採用時はモデルやフレームワークの対応だけでなく、SDK、メモリー、通信、構成規模を含めて評価する。

参考資料

AWS Trainium (新しいタブで開きます)

用語の説明や活用例に関連する参考資料です。

READ & LEARN

用語を知ったら、活用を知る。

AI LABで、AIの動向や業務に取り入れるヒントを読む。

記事一覧を見る