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 | Version |
|---|---|
| Python | 3.10 |
Install from Source
Step 1: Clone the Repository
Clone the AgentJet repository from GitHub and navigate into the project directory:
Step 2: Install Dependencies
AgentJet supports multiple backbones. Currently we have verl and trinity (recommended).
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. To build faster, export MAX_JOBS=${N_CPU}, or ensure a healthy connection to GitHub to install pre-compiled wheels.
# 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. To build faster, export MAX_JOBS=${N_CPU}, or ensure a healthy connection to GitHub to install pre-compiled wheels.
# 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. To build faster, export MAX_JOBS=${N_CPU}, or ensure a healthy connection to GitHub to install pre-compiled wheels.
# 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
