tracel-ai/models

何を解決するか

このリポジトリは、Burnディープラーニングフレームワークを使用して実装されたディープラーニングモデルの集中型コレクションを提供します。開発者は、これらのモデルを元から実装する必要なく、さまざまなAIタスク用の事前構築されたモデルアーキテクチャにアクセスできます。

動作方法

このプロジェクトは、公式およびコミュニティ寄稿によるモデルのセットを整理しています。これらの実装はBurnフレームワークの上に構築されており、Rustベースのディープラーニングの利用を可能にします。テキスト、画像、音声など、幅広いモダリティをカバーしています。

対象ユーザー

Burnフレームワークを使用し、言語モデル、画像分類、音声認識などのタスク用に即時利用可能なモデルアーキテクチャが必要な開発者や研究者。

主な特徴

• Llama、ALBERT、ResNetなどのモデルの公式実装。
• Stable Diffusion、Whisper、RWKV v7など、コミュニティによる寄稿。
• 複数のモダリティに対応：テキスト（LLM、エンコーダ）、視覚（分類、オブジェクト検出）、音声（音声認識、音符検出）。
• MITおよびApache 2.0ライセンスの下で配布。