Trainium

Trainiumは、Amazon Web Services（AWS）が独自開発する機械学習モデルのトレーニング専用チップである。NVIDIA製GPUへの依存低減とコスト効率の改善を目的とし、AWSのクラウド上で提供される。初代Trainiumに続き、性能とメモリ帯域を強化したTrainium2、さらに次世代となるTrainium3が展開されており、大規模言語モデルの学習用インフラとして「Project Rainier」と呼ばれる大規模クラスターにも組み込まれている。

概要

TrainiumはAWSが自社データセンター向けに設計するASIC（特定用途向け集積回路）であり、汎用GPUに比べて特定のワークロードにおける電力効率や単位コストあたりの性能を高めることを狙っている。同社の推論向けチップ「Inferentia」と対をなす位置づけで、トレーニング処理に特化している点が特徴だ。AWSはCPUのGravitonシリーズとAIアクセラレータのTrainiumシリーズを両輪とし、クラウド基盤全体の自社シリコン化を進めている。

沿革

Trainiumは初代から性能改善を重ね、Trainium2で大幅な計算能力とメモリ帯域の強化が図られた。Trainium2を大量に接続した大規模クラスターは「Project Rainier」の名で知られ、AI企業の大規模学習需要に対応する基盤として位置づけられている。後継のTrainium3も開発が進められており、AWSは世代交代を通じて性能とコスト効率の両立を追求してきた。

技術的位置づけ

生成AIブームに伴い、学習・推論に用いるチップの調達コストと供給制約は業界全体の課題となっている。NVIDIAの次世代AIラックの製造原価がメモリ価格上昇などで大幅に増加しているとの指摘もあり、クラウド事業者にとって自社設計チップによるコスト管理の重要性は高まっている。AWSのTrainiumは、こうした環境下でMetaのMTIAやGoogleのTPUと同様に、汎用GPUに代わる用途特化型シリコンの一つとして位置づけられる。

主要な動向

2026年6月には、AWSが汎用CPU「Graviton5」搭載インスタンスの提供を開始し、前世代比で計算性能が最大25%向上したと発表された。この動きはAI処理そのものを担うTrainiumとは異なるが、AI時代のクラウド基盤全体を自社シリコンで構成するAWSの戦略の一環として位置づけられる。同じ2026年6月には、AlphabetがAnthropicに最大400億ドルを投資する一方、AnthropicはAmazonやBroadcomとも提携し、複数クラウド・複数チップベンダーを併用する戦略を取っていることが明らかになった。この文脈でTrainiumを用いるProject Rainierは、Anthropicにとって計算資源確保の選択肢の一つとなっている。またAnthropicは推論コスト削減のため英国スタートアップFractileとの交渉も進めており、AI企業が学習・推論の両面で複数のチップ供給元を並行検討する動きが強まっている。こうした背景には、NVIDIAの次世代AIインフラの製造コスト上昇や、MetaがBroadcomと連携して自社チップMTIAを大規模展開する動きなど、業界全体で汎用GPUからの脱却が進んでいる状況がある。Trainiumも、この用途特化型シリコンへの潮流の中で、AWSのクラウド顧客に対する計算資源提供の選択肢として重要性を保っている。