siiRL

siiRL

siiRL: Shanghai Innovation Institute RL Framework for Advanced LLMs and Multi-Agent Systems

Stars: 199

Visit
 screenshot

siiRL is a novel, fully distributed reinforcement learning (RL) framework designed to break the scaling barriers in Large Language Models (LLMs) post-training. Developed by researchers from Shanghai Innovation Institute, siiRL delivers near-linear scalability, dramatic throughput gains, and unprecedented flexibility for RL-based LLM development. It eliminates the centralized controller common in other frameworks, enabling scalability to thousands of GPUs, achieving state-of-the-art throughput, and supporting cross-hardware compatibility. siiRL is extensively benchmarked and excels in data-intensive workloads such as long-context and multi-modal training.

README:



siiRL: Shanghai Innovation Institute RL Framework for Advanced LLMs and Multi-Agent Systems

| πŸ“„ Paper | πŸ“š Documentation | Feishu Group QR Code Feishu Group | Wechat Group QR Code Wechat Group | πŸ‡¨πŸ‡³ δΈ­ζ–‡ |

siiRL is a novel, fully distributed reinforcement learning (RL) framework designed to break the scaling barriers in LLM post-training. Developed by researchers from Shanghai Innovation Institute, siiRL tackles the critical performance bottlenecks that limit current state-of-the-art systems.

By eliminating the centralized controller common in other frameworks, siiRL delivers near-linear scalability, dramatic throughput gains, and unprecedented flexibility for RL-based LLM development.


πŸš€ Highlights

  • Near-Linear Scalability: The multi-controller paradigm eliminates central bottlenecks by distributing control logic and data management across all workers, enabling near-linear scalability to thousands of GPUs.

  • SOTA Throughput: Fully distributed dataflow architecture minimizes communication and I/O overhead, achieving SOTA throughput in data-intensive scenarios.

  • Flexible DAG-Defined Pipeline: Decouple your algorithmic logic from the physical hardware. With siiRL, you can define complex RL workflows as a simple Directed Acyclic Graph (DAG), enabling rapid, cost-effective, and code-free experimentation.

  • Cross-Hardware Compatibility: siiRL now officially supports Huawei's Ascend NPUs, providing a high-performance alternative for training and inference on different hardware platforms.

  • Proven Performance & Stability: Extensively benchmarked on models from 7B to 72B, siiRL delivering excellent performance across a wide range of tasks. Its advantages are particularly evident in data-intensive workloads such as long-context and multi-modal training.


πŸ“° News

  • [2025/09]:siiRL now integrates Megatron training backend with support for MoE training. Performance has been validated on Qwen3-MoE models (30B, 235B).

  • [2025/09]:siiRL now supports stable scaling on GPU clusters from 32 GPUs up to 1024 GPUs, with over 90% linear scalability efficiency, through collaboration with major manufacturers including Huawei Ascend, MetaX, and Alibaba Cloud.

  • [2025/09]:siiRL supports multi-turn interactions among multi-agents with the environment.

  • [2025/07]:siiRL adds MARFT support for LaMAS, enabling RL fine-tuning of multi-LLM agents via Flex-POMDP.

  • [2025/07]: siiRL now supports CPGD, a novel algorithm that enhances RL training stability and performance by regularizing large policy updates.

  • [2025/07]: We are excited to release siiRL to the open-source community! Check out our paper for a deep dive into the architecture and evaluation.


πŸ’‘ Architecture Overview

siiRL is a fully distributed RL framework designed for scalability on large-scale clusters. siiRL employs a multi-controller paradigm that uniformly dispatches all computational and data flow across each GPU. siiRL consists of three main components: a DAG Planner, DAG Workers, and a Data Coordinator.

Overview of siiRL

Figure 1. Overview of siiRL.

siiRL solves this problem with a fully distributed, multi-controller architecture.

Key components include:

  • DAG Planner: Translates a user-defined logical workflow (DAG) into a serialized, executable pipeline for each worker.
  • DAG Workers: The core execution units, with each worker bound to a single GPU, running its assigned tasks independently.
  • Data Coordinator: A set of distributed components (Distributed Dataloader and Distributed Databuffer) that manage the entire data lifecycle, from initial loading to intermediate data redistribution, without a central coordinator.

πŸ§ͺ Experiment

We conducted a comprehensive evaluation of siiRL's performance and scalability across various scenarios, comparing it with the SOTA RL framework, verl. The experiments demonstrate that siiRL exhibits outstanding performance across all metrics.

End-to-End Throughput

Under the standard PPO and GRPO algorithms, siiRL's throughput comprehensively surpasses the baseline. Particularly with the more data-intensive GRPO algorithm, siiRL effectively resolves data bottlenecks through its fully distributed architecture, achieving up to a 2.62x performance improvement.

PPO Algorithm Performance Comparison
Figure 2: End-to-end performance comparison using the PPO algorithm

GRPO Algorithm Performance Comparison
Figure 3: End-to-end performance comparison using the GRPO algorithm

Large-Scale Scalability

siiRL demonstrates near-linear scalability, smoothly extending up to 1024 GPUs. In contrast, the baseline framework fails under identical conditions due to OOM errors caused by its single-point data bottleneck. At the maximum batch size the baseline can support, siiRL's performance advantage can be as high as 7x.

siiRL Scalability Test
Figure 4: Near-linear scalability of siiRL on VLM models

VLM Task Performance Comparison
Figure 5: VLM task performance comparison under the baseline's maximum load

Long-Context Performance

When processing long-context tasks, data transfer overhead becomes a major bottleneck. siiRL's distributed dataflow design allows its performance advantage to become more pronounced as context length increases, achieving up to a 2.03x throughput improvement and successfully running a 72B model long-context task that the baseline could not handle.

Long-Context Performance Comparison
Figure 6: Performance comparison in long-context scenarios

Model Convergence

Experiments confirm that siiRL's performance optimizations do not come at the cost of model accuracy. With identical hyperparameters, siiRL's reward and entropy convergence curves are identical to the baseline's, while reducing the total training time by 21%.

Convergence Curve Comparison
Figure 7: Model convergence curve comparison


πŸ“š Resources

Documentation


πŸ—“οΈ Future Plans

siiRL is under active development. We are excited about the future and are focused on extending the framework's capabilities in two key directions: advancing multi-agent support and optimizing the base framework.

Advanced Multi-Agent Capabilities

Our flexible DAG-based design provides a natural and powerful foundation for complex multi-agent systems.

Support Training for VLA

The goal is to create an end-to-end, distributed RL solution capable of training large-scale, multi-modal VLA models for embodied AI tasks. You can track the development progress for this feature in this Pull Request

Base Framework Enhancements

We are continuously working to improve the performance, efficiency, and scalability of the base system.

We welcome community contributions! Please see our Contributing Guide to get started.


πŸ™ Acknowledgement

We would first like to thank the open-source RL framework verl, which we used as a primary baseline for our evaluations. We would like to directly acknowledge its hierarchical API design; we reuse the 3DParallelWorker base class from verl to manage system components in siiRL.

siiRL is also built upon a foundation of other great open-source projects. We would like to thank the teams behind PyTorch, Ray, vLLM, vLLM-Ascend and SGLang for their incredible work.

Our work aims to address the scalability challenges identified during our research, and we hope siiRL can contribute positively to the community's collective progress.


πŸ–‹οΈ Citation

If you find siiRL useful in your research, please consider citing our paper.

@misc{wang2025distflowfullydistributedrl,
      title={DistFlow: A Fully Distributed RL Framework for Scalable and Efficient LLM Post-Training}, 
      author={Zhixin Wang and Tianyi Zhou and Liming Liu and Ao Li and Jiarui Hu and Dian Yang and Jinlong Hou and Siyuan Feng and Yuan Cheng and Yuan Qi},
      year={2025},
      eprint={2507.13833},
      archivePrefix={arXiv},
      primaryClass={cs.DC},
      url={https://arxiv.org/abs/2507.13833}, 
}

For Tasks:

Click tags to check more tools for each tasks

For Jobs:

Alternative AI tools for siiRL

Similar Open Source Tools

For similar tasks

For similar jobs