Installation

Quick Install

pip install ensemble_launcher

Or from source:

git clone https://github.com/argonne-lcf/ensemble_launcher.git
cd ensemble_launcher
pip install .

Editable Install (Development)

pip install -e ".[dev-core]"

This includes pytest, pytest-timeout, pytest-asyncio, and mpi4py.

For the full dev environment (adds MCP and inference dependencies):

pip install -e ".[dev-extensions]"

Optional Extras

HPC (MPI)

For distributed multi-node execution:

pip install "ensemble_launcher[hpc]"

Installs mpi4py. Requires an MPI implementation (e.g. MPICH, OpenMPI) on the system.

MCP Support

For hosting MCP servers on HPC compute nodes:

pip install "ensemble_launcher[mcp]"

Installs MCP and paramiko for SSH tunneling.

Inference

For vLLM-based inference serving:

pip install "ensemble_launcher[inference]"

Installs vLLM, uvloop, uvicorn, and FastAPI.

Everything

pip install "ensemble_launcher[all]"

Installs HPC, MCP, and inference dependencies together.