oumi-ai/oumi

What it solves

Oumiは、基盤モデルのライフサイクル全体に関連する複雑さやボイラープレートコードを排除するために設計されたオープンソースプラットフォームです。統一された API とすぐに使えるレシピ（recipes）を提供し、データ準備、学習、評価、デプロイメントを処理するため、開発者はカスタムの学習ループやデータパイプラインを記述することなく、研究からプロダクションへ移行できます。

How it works

Oumiは、一連の CLI ツールと設定ファイルを通じて、モデル開発プロセスを効率化します。ユーザーは oumi train、oumi evaluate、および oumi infer コマンドと YAML 設定ファイル（recipes）を組み合わせて、特定のワークフローを実行できます。SFT, LoRA, QLoRA, GRPO などの幅広い学習技術をサポートし、vLLM や SGLang といった人気の推理エンジンと統合されています。さらに、oumi launch コマンドを使用して、AWS, Azure, GCP, および Lambda にといったクラウドプラットフォーム上で学習および推理ジョブをリモートでデプロイプロイメントできます。

Who it’s for

ML 研究者やエンタープライズチーム向けに構築されています。ローカルのノート PC から大規模なクラウドクラスターまで、さまざまな環境において、最先端の基盤モデル（10M から 405B パラメータ）を構築、微調整、デプロイメントすることが可能です。

Highlights

• End-to-End Lifecycle: LLM-as-a-Judge を用いたデータ合成および策定から、最終的なデプロイメントまでをカバーします。
• Zero Boilerplate: Llama, Qwen, および DeepSeek などの人気のあるモデルファミリーに対して、事前設定済みのレシピ（recipes）に対して提供します。
• Multi-Cloud Support: AWS, Azure, GCP, および Lambda 上でジョブを起動するネイティブな機能を提供します。
• Broad Model Support: テキストのみのモデルとマルチモーダルモデルの両方に互換性があります。
• SOTA Training Techniques: 分散学習（FSDP, DeepSpeed, DDP）および GRPO のような高度な微調整手法をネイティブサポートしています。