Skip to content

Repository files navigation

AxonX

An agent-native harness for quantitative research.

Python 3.12+ PyPI version PyPI monthly downloads GitHub monthly commit activity Apache License 2.0 CLI and MCP access AxonX documentation Read in English 阅读简体中文 Ask DeepWiki about AxonX

What is AxonX?

AxonX connects research code, task execution, logs, and results in one workspace. It represents data processing, factor analysis, training, prediction, and backtesting as Tasks with explicit input and output contracts. CLI, AxonX Studio, and Agent access the same capabilities through Job interfaces.

Researchers can inspect how a result was produced, reuse upstream data, compare experiments, and let an Agent investigate tasks and artifacts. Plugin authors supply the research algorithms; AxonX provides the execution and inspection infrastructure. AxonX is currently in alpha.

Why AxonX?

  • Reuse research code as Tasks. Typed inputs and outputs make data, model, and artifact requirements explicit. → Task contracts
  • Execution you can inspect. Submit Tasks to independent worker processes and follow status, progress, logs, and results. → Task management
  • Trace results back to their inputs. Workspace records keep parameters, artifacts, and upstream Task IDs together so you can reuse datasets and inspect experiment differences. → Task lineage
  • One workflow across CLI, AxonX Studio, and Agent. Use scripts, browser forms and charts, or an assistant that reads task evidence through configured tools. → AxonX Studio · Agent
  • Extend and run remotely. Package research capabilities as plugins and explicitly choose a remote execution target. → Plugins · Remote machines

AxonX Studio

AxonX Studio provides task submission, run details, machine resources, workspace browsing, and research result views.

AxonX Studio home

See Getting started with AxonX Studio for setup and connection instructions.

Quick start

Requires Python 3.12+, with local Task execution on macOS and Linux. Use an activated virtual environment.

Install from PyPI

pip install "axonx[studio]"

This includes the CLI, API, MCP, and prebuilt AxonX Studio. For the core alone, install axonx.

Install from source

Requires Node.js 22.13+ (22.x), 24.x, or 26+ to build AxonX Studio:

git clone https://github.com/FlowLLM-AI/AxonX.git
cd AxonX
pip install -e .
cd axonx_studio
npm ci && npm run build
cd ..
pip install ./axonx_studio

This installs the core from source and builds and installs AxonX Studio locally. See Contributing for development setup and AxonX Studio development to modify the frontend.

Configure .env

Create .env in the directory where you start AxonX. The CLI loads it automatically; existing environment variables take precedence.

# Service authentication: choose your own token
AXONX_SERVICE_TOKEN=replace-with-your-local-service-token

# Optional: built-in Agent (Claude-compatible backend)
# CLAUDE_CODE_API_KEY=your-api-key
# CLAUDE_CODE_BASE_URL=https://api.anthropic.com
# CLAUDE_CODE_MODEL_NAME=your-model-name

# Optional: Tushare data download
# AXONX_TUSHARE_TOKEN=your-tushare-token
# AXONX_TUSHARE_BASE_URL=http://api.waditu.com/dataapi

The service token is sufficient for task management and AxonX Studio. Fill in the Agent settings when using the research assistant, or the Tushare token when downloading market data; override the Tushare URL only for a compatible custom endpoint. See example.env for remote-service and DingTalk settings. Keep .env out of version control.

Open AxonX Studio

axonx start --service.host 127.0.0.1

Open http://127.0.0.1:1024/ and enter AXONX_SERVICE_TOKEN from .env in Settings → Service token. See the full quick start for asynchronous submission, waiting, and result inspection.

Research with an Agent

The built-in assistant uses the Claude Agent SDK and configured Job tools to inspect task status, logs, upstream relationships, and workspace artifacts. Configure the Agent backend, then open Agent in AxonX Studio. Ordinary research Tasks can run without model credentials.

Give it specific Task IDs and questions, for example:

  • “Check Task <task_id>'s status, tail logs, and upstream tasks. Explain where it failed and what to inspect next.”
  • “Compare Backtest Tasks <A> and <B>: check their common date window and cost assumptions before explaining the results.”

External Agents can also connect to the service's Streamable HTTP MCP endpoint at http://127.0.0.1:1024/mcp using the service's Bearer token. Available tools depend on the service configuration. See Agent usage and MCP integration for tools and permissions.

Repository skill

The repository includes an AxonX skill for Agents working with research plugins, Task execution, logs, and artifacts. It references the maintained development and operations guide.

In an Agent session with access to this checkout, ask: “Read skills/axonx/SKILL.md and use it to inspect Task <task_id>.” Automatic discovery depends on the Agent's skill configuration; adding this directory does not automatically load it into AxonX Studio's built-in assistant. Keep the source checkout available because the skill references repository documents.

How it works

CLI / AxonX Studio / external Agent → Job interfaces → Task execution → workspace records and artifacts.

Jobs validate calls and coordinate framework capabilities. For research submission, the TaskManager starts a worker process and returns a run identifier; the Task writes status, logs, and outputs. Query Jobs and AxonX Studio then read those records. axonx exec runs a Task directly in the current process.

Research plugins supply the algorithms. Upstream Task IDs record relationships, while users or scripts organize stage-by-stage execution. See the architecture for the component boundaries.

Quantitative research and plugins

AxonX research and execution overview

A typical research chain is raw data → ETL → training → prediction → backtesting, with factor analysis branching from ETL. Tasks record upstream IDs through source_tasks, so datasets and predictions can be reused across experiments. Users or calling programs submit each stage.

The core supplies Task contracts and runtime infrastructure. Research plugins implement factors, models, and backtesting logic:

Plugin source Purpose
Alpha158 Alpha158 research task implementations
Alpha158 Enhanced Extended Alpha158 research tasks and experiment guidance

Install a research plugin in the execution service's Python environment, then restart the service:

pip install axonx-alpha158
# Or: pip install axonx-alpha158-enhanced

Start with the research workflow for plugin installation and data prerequisites. Market-data and Agent features need their own provider configuration. See Interpreting backtests for return definitions, costs, and trading assumptions.

AxonX Studio reads the resulting artifacts to display training metrics, predictions, and backtest summaries. For example, the backtest view shows overall signal metrics alongside period summaries:

AxonX Studio backtest overall metrics

This screenshot illustrates an existing experiment's result view. See Research results for how to read each stage's outputs.

Documentation

I want to… Guide
Install and run my first Task Quick start
Submit and inspect tasks in a browser AxonX Studio
Run the research chain and understand results Research workflow
Configure the research assistant or external MCP tools Agent configuration · MCP integration
Execute on another machine Remote machines
Configure and call AxonX Configuration · CLI · Python
Build research tasks or framework extensions Development guide · Framework extensions

Browse the complete bilingual documentation.

Contributing

Bug reports, feature requests, documentation improvements, research plugins, and code contributions are welcome. Search existing issues and read the contribution guide for development setup and checks.

License

AxonX is released under the Apache License 2.0.

About

Agent-native harness for quantitative research, with plugin-based tasks, traceable artifacts, and CLI, Studio, and MCP interfaces.

Topics

Resources

Contributing

Stars

1 star

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages