Teaching LLM committees to think in trees
MARTI trains squads of LLM agents with reinforcement learning, using tree search to collaboratively crack complex reasoning tasks.

What it does MARTI is a training framework that treats a network of LLM agents as a single reinforcement-learning problem. It centralizes how agents interact and share rewards while distributing the actual policy training across individual models. The newer MARTI-v2 layer adds Monte Carlo tree search to the mix, targeting tasks like code generation where multi-step exploration matters.
The interesting bit Instead of just chaining prompts, MARTI actually trains the agents’ weights through RL algorithms like GRPO and REINFORCE++, and it can simultaneously fine-tune heterogeneous models—say, a Qwen3-8B alongside a different 8B model—with independent roles and per-agent sample filtering. The tree-search variant (MARS²) swaps token-level PPO for sequence-level GSPO loss and applies a “TIS correction” to compensate for vLLM sampling drift during long rollouts.
Key highlights
- Supports graph-based workflows (debate, chain-of-agents, mixture-of-agents) and third-party integrations with AutoGen and CAMEL.
- MARTI-v2 adds single- and multi-agent MCTS training for code generation, with adaptive node expansion and refinement.
- Handles heterogeneous multi-agent training where different models run on independent strategies and dynamic sample filters.
- Incorporates sequence-level GSPO loss and TIS correction for stable training on sequences up to 32K tokens.
- Built on top of OpenRLHF and verl, using the vLLM v1 Engine and a Hybrid Engine for throughput.
Caveats
- The README lists steep hardware requirements: roughly eight 80GB GPUs for single-agent runs, and a full 8-GPU node per agent for multi-agent training.
- Third-party framework integrations (AutoGen, CAMEL) are marked as experimental.
- Reported benchmark gains are specific to the LCB code-generation dataset; broader task coverage isn’t shown in the sources.
Verdict Worth a look if you’re running a multi-agent LLM research lab and have the GPU budget to train agent teams end-to-end. If you’re looking for a lightweight CPU-friendly orchestration layer, this is not your tool.
Frequently asked
- What is TsinghuaC3I/MARTI?
- MARTI trains squads of LLM agents with reinforcement learning, using tree search to collaboratively crack complex reasoning tasks.
- Is MARTI open source?
- Yes — TsinghuaC3I/MARTI is open source, released under the MIT license.
- What language is MARTI written in?
- TsinghuaC3I/MARTI is primarily written in Python.
- How popular is MARTI?
- TsinghuaC3I/MARTI has 538 stars on GitHub.
- Where can I find MARTI?
- TsinghuaC3I/MARTI is on GitHub at https://github.com/TsinghuaC3I/MARTI.