Installation

Previous Next

This document provides a step-by-step guide to installing AgentJet.

Latest Version Recommended:

AgentJet is under active development and iteration. We recommend installing from source to get the latest features and bug fixes.

Prerequisites

Requirement Detail
Python 3.10
Package Management uv or conda

Install from Source

Step 1: Clone the Repository

Clone the AgentJet repository from GitHub and navigate into the project directory:

git clone https://github.com/modelscope/AgentJet.git
cd AgentJet

Step 2: Install Dependencies

AgentJet supports multiple backbones, you can choose any of them depending on your requirements, or choose all of them to compare the performance. Currently we have verl and trinity.

Package Manager

We recommend using uv to manage your Python environment as it is incredibly fast. See also uv installation document.

And of course, if you prefer conda, you can also install via conda and pip (simply change uv pip to pip).

# Install with `verl` training backbone:

uv venv --python=3.10
source .venv/bin/activate
uv pip install -e .[verl]

#`flash-attn` must be installed after other dependencies
uv pip install --verbose flash-attn --no-deps --no-build-isolation --no-cache

flash-attn Installation

  • flash-attn must be installed after other dependencies.
  • If you find your machine spend a long time installing flash-attn, ensure a healthy connection to GitHub.
# Install with `verl` training backbone:

conda create -n ajet-verl python=3.10
conda activate ajet-verl
pip install -e .[verl]

#`flash-attn` must be installed after other dependencies
pip install --verbose flash-attn --no-deps --no-build-isolation --no-cache

flash-attn Installation

  • flash-attn must be installed after other dependencies.
  • If you find your machine spend a long time installing flash-attn, ensure a healthy connection to GitHub.
# Install with `verl` training backbone:

uv venv --python=3.10
source .venv/bin/activate
uv pip install -i https://mirrors.aliyun.com/pypi/simple/ -e .[verl]

#`flash-attn` must be installed after other dependencies
uv pip install -i https://mirrors.aliyun.com/pypi/simple/ --verbose flash-attn --no-deps --no-build-isolation --no-cache

flash-attn Installation

  • flash-attn must be installed after other dependencies.
  • Ensure a healthy connection to GitHub to install pre-compiled wheels.
  • If you find your machine spend a long time installing flash-attn, ensure a healthy connection to GitHub.
  • To build faster, export MAX_JOBS=${N_CPU}.
# Install with `trinity` training backbone for fully asynchronous RFT:

uv venv --python=3.10
source .venv/bin/activate
uv pip install -e .[trinity]
uv pip install --verbose flash-attn --no-deps --no-build-isolation --no-cache

# Install with `trinity` training backbone for fully asynchronous RFT:

uv venv --python=3.10
source .venv/bin/activate
uv pip install -i https://mirrors.aliyun.com/pypi/simple/ -e .[trinity]
uv pip install -i https://mirrors.aliyun.com/pypi/simple/ --verbose flash-attn --no-deps --no-build-isolation --no-cache
Backbone VeRL Trinity-RFT
Core design Share-GPU actor-rollout engine (colocate) Async actor-rollout engine
Speed ⭐⭐⭐⭐ ⭐⭐⭐⭐
Scalability ⭐⭐ ⭐⭐⭐⭐
Minimum Required GPU Resource 1 2
Training Stability ⭐⭐⭐⭐ ⭐⭐⭐
vLLM Version 0.10.0 0.10.0

Next Steps

Quick Start

Run your first training command and explore examples.

Tune Your First Agent

Step-by-step guide to build and train your own agent.
Introduction Quick Start