EN

Oumi:面向基础模型的开源平台

Oumi 是完全开源的平台,覆盖基础模型从数据准备、训练到评估和部署的生命周期。

核验日期

核心信息

以下信息由程序从官方接口读取,不经模型改写。

许可证
Apache-2.0
语言
Python
维护方
oumi-ai
核验日期

它是什么

Oumi 是一个完全开源的平台,简化基础模型从数据准备、训练到评估和部署的整个生命周期。项目称 Oumi 是社区优先的工作。该项目采用 Apache License 2.0 许可。

主要能力

  1. Oumi 可以处理文本模型和多模态模型,包括 Llama、Gemma、Qwen、DeepSeek、GLM、Phi 等。

  2. Oumi 可以合成和整理训练数据,包括多轮工具调用对话,并使用 LLM 评判器。

  3. Oumi 可以通过 deploy 命令把模型部署到专用端点。

  4. Oumi 可以在标准基准上评估模型,也可以使用多标准的评分细则评判器。

  5. Oumi 提供可执行的工具环境,包括数据库、HTTP 端点、查找和模拟环境。

怎么上手

  1. 可以使用 launch 命令在云平台(AWS、Azure、GCP、Lambda 等)上远程运行任务。

  2. 安装方式中列有使用 pip,并标注为推荐。

  3. 安装方式中也列有使用 Docker。

  4. 另有一个标注为实验性的快速安装脚本。

摘自 README · bash
# Basic installation
uv pip install oumi

# With GPU support
uv pip install 'oumi[gpu]'

# Latest development version
uv pip install git+https://github.com/oumi-ai/oumi.git
摘自 README · bash
# Pull the latest image
docker pull ghcr.io/oumi-ai/oumi:latest

# Run oumi commands
docker run --gpus all -it ghcr.io/oumi-ai/oumi:latest oumi --help

# Train with a mounted config
docker run --gpus all -v $(pwd):/workspace -it ghcr.io/oumi-ai/oumi:latest \
    oumi train --config /workspace/my_config.yaml
摘自 README · bash
curl -LsSf https://oumi.ai/install.sh | bash
摘自 README · shell
# Training
oumi train -c configs/recipes/smollm/sft/135m/quickstart_train.yaml

# Evaluation
oumi evaluate -c configs/recipes/smollm/evaluation/135m/quickstart_eval.yaml

# Inference
oumi infer -c configs/recipes/smollm/inference/135m_infer.yaml --interactive
摘自 README · shell
# GCP
oumi launch up -c configs/recipes/smollm/sft/135m/quickstart_gcp_job.yaml

# AWS
oumi launch up -c configs/recipes/smollm/sft/135m/quickstart_gcp_job.yaml --resources.cloud aws

# Azure
oumi launch up -c configs/recipes/smollm/sft/135m/quickstart_gcp_job.yaml --resources.cloud azure

# Lambda
oumi launch up -c configs/recipes/smollm/sft/135m/quickstart_gcp_job.yaml --resources.cloud lambda

适合谁

Oumi 处于 beta 阶段,并在积极开发中。README 中的示例配置并不是所支持内容的完整列表。README 把开放模型定义为权重、训练代码和数据完全开放且许可宽松的模型。

Oumi 的运行范围从笔记本电脑到集群再到云(AWS、Azure、GCP、Lambda、Modal、Slurm 等)。Oumi 可以与开放模型和商业 API 集成,包括 OpenAI、Anthropic、Vertex AI、Together、Fireworks、Parasail 等。oumi-mcp MCP 服务器可以让用户从 Claude 或 Cursor 驱动 Oumi。

项目自称其“零样板”特性意味着无需编写训练循环或数据管道。Oumi 对分布式训练技术(FSDP、DeepSpeed、DDP)提供原生支持。项目自称 Oumi 由大规模训练模型的团队构建并验证。

更新记录中提到,GRPO 强化学习已扩展到支持工具使用。更新记录中提到,已添加对 Gemma 4 模型系列的支持。更新记录中提到,新的 deploy 命令可把 Oumi 模型部署到 Fireworks 和 Parasail 上的专用推理端点。

更新记录中提到,已添加对 Qwen3.5 模型系列的支持。更新记录中提到,支持 OpenAI 的 gpt-oss-20b 和 gpt-oss-120b 的训练与推理。更新记录中提到,已添加对 Llama 4 模型训练和推理的支持。

更新记录中提到,已添加对工具、环境(模拟、查找、数据库)和智能体数据合成的支持。

来源

本页内容整理自以下官方来源,点击可核对原文。

核验日期 ·

本页由 AI 依据上述官方来源整理,并经自动核对,未经本站实测。 发现错误?告诉我们。