Oossa

Amazon、SGLang-NeuronをAWS Trainium向けに提供開始

AmazonがSGLang-Neuronをパブリックコンテナとして提供。使い慣れたSGLangのワークフローで、Trainiumチップ上の大規模言語モデルを実行できる。

短信著者: Oossa公開日: 1 分で読める

Amazonは、AWS Deep Learning Containerを通じてSGLang-Neuronの一般提供を開始したと発表した。このプラグインは、SGLangのサービングシステムとAWS Trainium AIチップを連携させ、使い慣れたSGLang APIでモデルを起動し、RLロールアウトを生成できる。ネイティブのPyTorch上で動作し、Mixture of ExpertsやElastic Fabric Adapter経由のKVキャッシュ転送などの機能に対応する。クイックスタートガイドでは、Trainium3搭載のEC2インスタンスでコンテナを起動し、1,200億パラメーターのモデルをサービングする方法を紹介している。

なぜ重要か

開発者はSGLangのコードを変更せずに、AWSのTrainiumチップ上で非常に大規模な言語モデルを実行できる。これにより、コストを抑え、デプロイを簡素化できる。

この記事は役に立ちましたか?
共有

次に読む

Oossa · ニュースレター

今週のAIを、わかりやすく

毎週月曜日、知っておきたいニュースをやさしい言葉で。無料・スパムなし。