vime  by vllm-project

LLM post-training framework for scalable RL

Created 2 months ago
344 stars

Top 80.2% on SourcePulse

GitHubView on GitHub
Project Summary

Summary

Vime is an LLM post-training framework designed for Reinforcement Learning (RL) scaling, built upon the slime project and integrating the vLLM inference engine. It targets researchers and engineers seeking efficient LLM fine-tuning and flexible data generation workflows. Vime offers high-performance training by connecting Megatron with vLLM, enabling scalable RL applications.

How It Works

Vime's architecture comprises three core modules: training (Megatron), rollout (vLLM + router), and data buffer. The training module handles the primary optimization process, consuming data from the data buffer and synchronizing model parameters. The rollout module leverages vLLM for high-throughput inference, routing generation requests and producing new training data, including rewards, which is then fed back into the data buffer. This design facilitates efficient training and arbitrary data generation workflows by bridging Megatron's training stack with vLLM's inference capabilities.

Quick Start & Requirements

The README refers to a "Quick Start Guide" and "examples" sections for comprehensive details on environment setup, data preparation, and training startup. Specific installation commands (e.g., pip install) or explicit prerequisites like Python/CUDA versions are not detailed, though vLLM integration implies GPU hardware is necessary. Code style consistency is managed via pre-commit.

Highlighted Details

  • Supports a wide range of LLM architectures, including Qwen (3.6, 3.5, 3Next, 3MoE, 3, 2.5), DeepSeek V3 (V3, V3.1, R1), and Llama 3.
  • Enables high-performance training by connecting Megatron with vLLM for efficient RL scaling.
  • Provides flexible data generation capabilities through custom interfaces and server-based engines.
  • Maintained by the vLLM community, positioning itself as a production-ready bridge within the vLLM ecosystem.

Maintenance & Community

Vime is actively maintained by the vLLM community. Discussion channels include Slack and a WeChat group. Contributions are welcomed via Issues or Pull Requests, with pre-commit used for code style enforcement. The project builds upon and references documentation from the upstream slime repository.

Licensing & Compatibility

The project's license is not specified in the provided README. Consequently, compatibility for commercial use or closed-source linking cannot be determined from this document.

Limitations & Caveats

The README does not explicitly detail project limitations, alpha status, or known bugs. As Vime is derived from slime, users may need to consult slime's documentation for potential inherited caveats.

Health Check
Last Commit

18 hours ago

Responsiveness

Inactive

Pull Requests (30d)
82
Issues (30d)
15
Star History
129 stars in the last 30 days

Explore Similar Projects

Starred by Bryan Helmig Bryan Helmig(Cofounder of Zapier), Will Brown Will Brown(Research Lead at Prime Intellect), and
1 more.

ReCall by Agent-RL

0.3%
1k
RL framework for LLM tool use
Created 1 year ago
Updated 1 year ago
Starred by Chip Huyen Chip Huyen(Author of "AI Engineering", "Designing Machine Learning Systems"), Luis Capelo Luis Capelo(Cofounder of Lightning AI), and
3 more.

LitServe by Lightning-AI

0.2%
4k
AI inference pipeline framework
Created 2 years ago
Updated 5 days ago
Starred by Chip Huyen Chip Huyen(Author of "AI Engineering", "Designing Machine Learning Systems"), Wing Lian Wing Lian(Founder of Axolotl AI), and
3 more.

ROLL by alibaba

0.4%
3k
RL library for large language models
Created 1 year ago
Updated 1 day ago
Starred by George Hotz George Hotz(Author of tinygrad; Founder of the tiny corp, comma.ai), Eric Zhu Eric Zhu(Coauthor of AutoGen; Research Scientist at Microsoft Research), and
13 more.

slime by THUDM

1.4%
7k
LLM post-training framework for RL scaling
Created 1 year ago
Updated 3 days ago
Feedback? Help us improve.