⚡ AgentJIT
Just-In-Time Compiler for AI Agent Trajectories
Compile flaky, 30-second multi-step AI Agent workflows into 5-millisecond deterministic code.
🤝 Community, Issues & Support
- 🐛 Found a bug, compiler deopt issue, or guard failure? Please Open an Issue — reports are tracked and resolved quickly!
- 💬 Questions, trajectory compilation ideas, or new backends? Join our GitHub Discussions.
- ⭐ Find AgentJIT useful? Give it a star on GitHub — it helps more AI agent engineers discover the compiler!
- ☕ Support development (USDT TRC-20):
TDVbEdnpgNgoAhNcn1EwXxWHiR3RKLit5P(Verify on TronScan)
Quickstart • Compatibility • Why AgentJIT? • FASM Acceleration • Architecture • Benchmarks
🏆 Featured as #1 Top Story in The Daily Diff: Selected by Arpit Bhayani as the lead AI story (Depth: 9/10, Utility: 9/10, Novelty: 9/10, Interest: 9/10). Discussion on Hacker News.
🧩 Universal Compatibility Matrix
| Runtime / Implementation | Supported Versions | Execution Mode | Status |
|---|---|---|---|
| CPython (Standard) | 3.8, 3.9, 3.10, 3.11, 3.12, 3.13, 3.14, 3.15, 3.16 (Alpha) | Bytecode + GIL | ✅ Fully Supported |
| CPython (Free-Threaded) | 3.13t, 3.14t, 3.15t, 3.16t (Alpha) | Multi-core No-GIL (PEP 703) | ✅ Fully Supported |
| PyPy (JIT Accelerated) | 3.8, 3.9, 3.10, 3.11, 3.12 | High-speed JIT tracing | ✅ Fully Supported |
| FASM Hardware Engine | AVX2+FMA (x86-64), SSE2 (x86 32-bit) | Bare-Metal Assembly Microkernels | ✅ Fully Supported |
| Operating Systems | Windows (7, 8, 10, 11), Linux, macOS (Intel & Apple Silicon) | x86_64, ARM64 | ✅ Fully Supported |
💥 The Problem in 2026: Why AgentJIT?
In 2026, autonomous AI agents solve real-world workflows across business, DevOps, and data analysis. However, running stochastic LLM loops in production faces four critical barriers:
- Massive Latency: A standard 4-step agent workflow (
think -> tool -> observe -> think) takes 15 to 45 seconds. - Exponential Costs: Running the loop 10,000 times/day costs thousands of dollars in redundant API tokens.
- Flakiness & Hallucinations: Even 98% reliability per step leads to compounding errors across multi-turn trajectories.
- Redundant Reasoning: Most agent invocations execute the exact same structural trajectory with slightly different input parameters (e.g. different user IDs or dates).
⚡ Bare-Metal FASM Hardware SIMD Acceleration
AgentJIT integrates a high-performance Flat Assembler (FASM) hardware microkernel engine providing unrolled, native assembly acceleration for critical agent execution hot paths:
- 🏎️ AVX2 + FMA (x86-64) & SSE2 (x86 32-bit): Hand-crafted assembly microkernels with 256-bit SIMD registers (
agentjit64.dll,agentjit32.dll). - ⚡ Sub-Nanosecond Trajectory Hashing: 64-bit unrolled hash algorithms (
agentjit_fast_hash) for instantaneous prompt and trajectory cache lookups (> 300,000 hashes/sec). - 🛡️ Vectorized Speculative Guards: Parallel NULL-pointer evaluation and numerical range validation across parameters simultaneously using SIMD vector comparison.
- 🧠 Semantic Prompt Similarity Router: Unrolled 32-float AVX2+FMA cosine distance check (
agentjit_semantic_guard_cosine) comparing prompt embeddings in < 0.1 µs to speculatively route to compiled pipelines or de-optimize to the LLM agent. - 📦 Zero External C/C++ Compiler Dependencies: Assembled into standalone DLLs directly via FASM without MSVC, GCC, or bloated toolchains.
import agentjit
# Verify active hardware SIMD backend
print(f"Active Backend: {agentjit.simd_backend()}")
# -> "AVX2+FMA (FASM x86-64, 256-bit SIMD)"
# Access the hardware engine directly
engine = agentjit.get_fasm_engine()
h = engine.fast_hash("Process order 1042")
all_valid = engine.eval_range_guards([25.0, 150.0], [0.0, 100.0], [50.0, 500.0])
🏗️ Architecture & Trajectory JIT Compilation
Just like V8 compiles hot JavaScript into machine code, and PyTorch torch.compile traces dynamic tensors into optimized CUDA kernels, AgentJIT traces dynamic agent trajectories and compiles them into pure, type-safe, ultra-fast Python code.
[Dynamic Agent Task]
│
(1st run / warmup)
▼
┌──────────────────────┐
│ AgentJIT Tracer │ ── (Captures tool calls, data flow, variables)
└──────────────────────┘
│
▼
┌──────────────────────┐
│ DAG Flow Analyzer │ ── (Parameter generalization, dependency graph)
└──────────────────────┘
│
▼
┌──────────────────────┐
│ AST Code Generator │ ── (Synthesizes pure Python pipeline + Guards)
└──────────────────────┘
│
▼
┌────────────────────────────┐
│ Compiled JIT Pipeline │ ──► Subsequent runs: <1ms, $0 tokens!
└────────────────────────────┘
│
(Guard failure? Deopt!)
▼
[Fall back to LLM Agent]
⚡ Key Features
- 🏎️ Up to 100,000x Speedup: Hot paths drop from ~20,000 ms to < 0.1 ms.
- 💸 100% Token Savings: Once compiled, recurring workflows run completely locally with 0 LLM tokens consumed.
- 🛡️ Speculative De-Optimization (Bailout): Automatically generates runtime input guards. If unexpected data formats or divergent branches appear, AgentJIT transparently falls back to the dynamic LLM agent.
- 🔍 Transparent & Inspectable: Inspect the exact Python code generated by the JIT with
agent.source_code. - 🧵 Free-Threaded / No-GIL (PEP 703) Ready: Thread-safe runtime fully tested on Python 3.13t and 3.14t for true multi-core parallel agent execution without GIL contention.
- 🔌 Framework Agnostic: Seamlessly wraps LangChain, CrewAI, AutoGen, OpenAI Tool calls, or native Python functions.
🚀 Quickstart
Installation
| Platform / Manager | Installation Command |
|---|---|
| PyPI (pip) | pip install agentjit |
| PyPI (uv) | uv add agentjit |
| Conda (Anaconda.org) | conda install -c m_n_nik agentjit |
| Ubuntu / Debian (APT PPA) | curl -sS https://eminsk.github.io/ppa/setup.sh | sudo bashsudo apt install python3-agentjit |
| Ubuntu / Debian (.deb) | sudo dpkg -i python3-agentjit_0.1.6-1_all.deb |
pip install agentjit
# or with uv
uv add agentjit
10-Second Example
Decorate your agent with @jit and mark your tools with @trace_tool:
from agentjit import jit, trace_tool
# 1. Define your tools
@trace_tool()
def search_product(name: str):
return {"name": name, "price": 49.99, "stock": 120}
@trace_tool()
def apply_tax(price: float, tax_rate: float):
return round(price * (1.0 + tax_rate), 2)
# 2. Decorate your agent with @jit
@jit
def checkout_agent(product_name: str, tax_rate: float):
# This dynamic workflow could call an LLM (Claude, GPT, Gemini)
item = search_product(name=product_name)
total = apply_tax(price=item["price"], tax_rate=tax_rate)
return {"item": item["name"], "total": total}
# --- Run 1: Warmup & Tracing (runs dynamic agent, compiles to Python) ---
order1 = checkout_agent("Mechanical Keyboard", 0.19)
# --- Run 2+: Instant compiled execution (ZERO tokens, sub-millisecond!) ---
order2 = checkout_agent("Wireless Mouse", 0.19) # Takes 0.05 ms!
🔎 Inspecting Generated Code
You can view the exact synthesized Python code generated by the JIT at any time:
print(checkout_agent.source_code)
Synthesized Output:
def compiled_checkout_agent(product_name, tax_rate):
"""JIT-compiled trajectory pipeline generated by AgentJIT.
Executes deterministically in sub-millisecond time with zero token cost.
"""
# --- Speculative Guards ---
if not (product_name is not None):
raise GuardViolation("Argument 'product_name' must not be None", param="product_name")
if not (isinstance(product_name, str)):
raise GuardViolation("Argument 'product_name' must be of type str", param="product_name")
# --- Execution Steps ---
step_1_out = _tools['search_product'](name=product_name)
step_2_out = _tools['apply_tax'](price=step_1_out['price'], tax_rate=tax_rate)
# --- Return Final Result ---
return {'item': step_1_out['name'], 'total': step_2_out}
📊 Benchmarks
Benchmark comparing a simulated 3-step reasoning agent (15s latency, 2,500 tokens) vs AgentJIT compiled execution over 100 runs:
| Execution Mode | Mean Latency | 99th Percentile | Cost per 1k runs | Token Usage | Determinism |
|---|---|---|---|---|---|
| Standard LLM Agent | 14,820 ms |
22,400 ms |
$75.00 | 2,500,000 | ~94% |
| AgentJIT (Warm Path) | 0.08 ms |
0.12 ms |
$0.00 | 0 | 100% |
| Improvement | 185,000x faster | 186,000x faster | 100% savings | Zero tokens | Rock-solid |
🛡️ Speculative Execution & Bailouts
What happens when an input is unusual or triggers an unexpected branch?
AgentJIT uses Speculative De-Optimization:
- Input variables are validated against synthesized guards.
- If any guard fails (e.g. wrong type, missing required key) or a tool raises an unhandled exception, AgentJIT catches
GuardViolation. - It seamlessly bails out to the dynamic LLM agent to handle the edge case.
- Telemetry records the bailout for future multi-branch specialization.
# Normal input: runs compiled pipeline in 0.08ms
checkout_agent("Monitor", 0.19)
# Divergent input (e.g. invalid type): automatically bails out to dynamic agent
checkout_agent(12345, None) # Transparently de-optimizes, no crash!
🤖 Native Model Context Protocol (MCP) Server
AgentJIT provides a built-in Model Context Protocol (MCP) Server over JSON-RPC 2.0 stdio, empowering Claude Desktop, Cursor, Windsurf, and Google Antigravity to compile their own tool calling trajectories into deterministic Python functions on the fly.
Quickstart
Launch the MCP server via CLI:
agentjit mcp
# or using the direct entry point:
agentjit-mcp
Client Configuration
Add to your claude_desktop_config.json, .cursor/mcp.json, or Antigravity MCP settings:
{
"mcpServers": {
"agentjit": {
"command": "agentjit-mcp"
}
}
}
Or zero-install with uvx:
{
"mcpServers": {
"agentjit": {
"command": "uvx",
"args": ["agentjit", "mcp"]
}
}
}
Available MCP Tools
| Tool | Purpose | Annotations |
|---|---|---|
agentjit_compile |
Synthesizes deterministic Python code with input type guards from an agent tool trace. | readOnlyHint: true |
agentjit_analyze |
Analyzes DAG data-flow dependencies, identifying constant literals vs dynamic parameters. | readOnlyHint: true |
agentjit_simulate_savings |
Computes exact dollar cost and latency savings when compiling recurring LLM workflows. | readOnlyHint: true |
agentjit_info |
Returns runtime compiler capabilities, Free-Threaded (PEP 703 No-GIL), and PyPy JIT status. | readOnlyHint: true |
🛠️ Telemetry & Observability
Monitor your compiled agents in real time:
print(checkout_agent.stats)
# Output:
# {
# "total_calls": 1500,
# "compiled_hits": 1492,
# "bailouts": 8,
# "compiled_hit_rate": 99.47,
# "total_time_saved_ms": 22380000.0,
# "total_tokens_saved": 3730000
# }
🗺️ Roadmap for 2026–2027
- Core Tracer & DAG Flow Analyzer
- AST Code Generation with Speculative Guards
- De-optimization / Bailout Runtime
-
@jitDecorator with Auto-Warmup - Multi-Branch Polyhedral JIT: Merge multiple execution paths into a unified control-flow graph (
if/elsebranching synthesis). - eBPF-Isolated Micro-Sandbox: Ultra-fast sub-millisecond process sandbox for compiled shell actions.
- WebAssembly (Wasm) Export: Compile agent trajectories into standalone Wasm binaries for browser and edge runtime.
🧪 Testing & Verification
AgentJIT includes a comprehensive test suite covering the Tracer, Flow Analyzer, AST CodeGen, Runtime Bailout, and End-to-End decorators:
uv run --extra dev pytest -v
# or with standard pytest
pytest -v
All 9 test suites pass 100% across Python 3.8 through 3.16 (including No-GIL free-threaded 3.13t–3.16t) and PyPy 3.8 through 3.12.
🌐 High-Performance Systems Ecosystem
agentjit is developed by @eminsk as part of an open-source AI & systems engineering ecosystem:
- ⚡ NanoVector — Bare-metal C99/AVX2 vector search & episodic memory engine (~120KB) with Native MCP Server (
pip install nanovector). - ⚡ NanoGEMM — Bare-metal AVX2+FMA SIMD matrix multiplication engine in ~100KB for sub-microsecond CPU inference (
pip install nanogemm). - 🖥️ NanoRecall — 100% Private, offline desktop memory & semantic screen search engine powered by NanoVector (
pip install nanorecall). - 🛒 avito-sdk — Headless Avito scraping & data extraction SDK with price drop tracking, Playwright cookies, Telegram/VK bots, and Native MCP Server (
pip install avito-sdk). - 📈 yfinance-ta-patterns — Institutional-grade technical pattern scanner with AI Confluence Scoring, Backtester, and Native MCP Server (
pip install yfinance-ta-patterns). - 📊 xlsx_vievers — Headless Excel formula engine (129+ functions), desktop spreadsheet viewer, SIMD SSE2 math, and Native MCP Server (
pip install xlsx-viewer-pro). - 🎥 screenvideo — Desktop screen recorder with WASAPI audio and standalone pure x64 FASM edition.
☕ Support, Community & Donations
If you find this project valuable and would like to support ongoing development:
- ⭐ Star the Repository: If AgentJIT helps your AI agents run faster, give us a star on GitHub — it helps more developers discover the project!
- 💬 Join Discussions: Have ideas, use cases, or questions? Start or join a thread in GitHub Discussions!
- ☕ Donate (USDT TRC-20):
TDVbEdnpgNgoAhNcn1EwXxWHiR3RKLit5P
(Network: TRON / TRC-20 | Verify on TronScan)
📄 License
AgentJIT is open-source software licensed under the Apache 2.0 License.
Metadata
Release files for agentjit 0.1.9
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| agentjit-0.1.9.tar.gz | 927.6 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| agentjit-0.1.9-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 969.1 kB
Release files / agentjit-0.1.9.tar.gz
| Download URL | agentjit-0.1.9.tar.gz |
|---|---|
| Size | 927.6 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
da572058814d97cd2e176ad7d57d8310d9513dc59d2b184989a9617daffd6ae4
|
|
BLAKE2b-256 checksum How to use checksums |
a96359a715960df8d817fce2a63c9590cb2818d0349fb3aecff60ba181430d02
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Release files / agentjit-0.1.9-py3-none-any.whl
| Download URL | agentjit-0.1.9-py3-none-any.whl |
|---|---|
| Size | 41.5 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
da492eedfa844814bc05af4e6be5fe811ca6712d3f3e363897f13cda8ea7defb
|
|
BLAKE2b-256 checksum How to use checksums |
a8f4342426483e9684f45e1007ed9e2f4ceb581c485688be63882ae4bd269bfe
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|