Documentation
Introduction
What aimake is, why AI pipelines need incremental builds, and how it fits next to Make, DVC, and orchestrators.
What is aimake?
aimake is an incremental build system for AI and ML pipelines. It sits in the same mental model as make, git, and DVC — but it is shaped for the artifacts AI teams actually touch: datasets, prompts, embeddings, indexes, models, evaluations, and reports.
Install from PyPI (pip install aimake). Source and issues live on GitHub. Current stable line is v2.0.
When only a prompt changes, everything upstream should be skipped. aimake tracks a dependency DAG, fingerprints inputs with SHA-256 content hashes (not file mtimes), and rebuilds only the nodes that actually changed — locally, in CI, or across a shared remote cache.
Why traditional tools fall short
Classic build tools understand source → object → binary. AI pipelines look different:
dataset
│
▼
preprocess
│
▼
embeddings
│
▼
index ─────────────┐
│
prompt ────────────┼──► evaluation
│
▼
report
| Tool | Focus |
|---|---|
| Make | Generic file dependencies (mtime-based) |
| DVC | Data and model versioning |
| MLflow | Experiment tracking |
| Prefect / Airflow | Scheduling and orchestration at scale |
| aimake | Incremental AI pipeline builds with content-addressable caching |
aimake does not try to replace orchestrators or data registries. It answers a narrower question: given this DAG of AI artifacts, what must run again, what can be restored from cache, and what will it cost?
See the full comparison in Comparison.
What you get
| Category | Capabilities |
|---|---|
| Core | Dependency DAG, SHA-256 fingerprinting, parallel builds, content-addressable cache |
| CLI | 25+ commands for build, plan, inspect, explain, diff, compare, optimize, registry |
| Cache | Local SQLite + filesystem; optional S3 remote (push / pull / sync) |
| Compute | GPU-aware scheduling, distributed SSH workers |
| Experiments | Grid / random / Bayesian / Optuna search, Hyperband pruning, Pareto multi-objective |
| Integrations | MLflow, Hugging Face Hub, W&B, DVC, Docker, Ollama, artifact registry |
| CI | Quality gates, doctor health checks, official GitHub Action, eval --check |
| Team | Shared remote cache, aimake.lock, registry promote policies, schedules, notifications |
| Trust | External probes, attestation, reproducibility reports, lineage export |
The mental model in one minute
- You declare artifacts in
aimake.yaml— each withdepends_on,command,outputs, and optionalmetrics,external, andvalidation. aimake planshows which steps will run, restore, or skip — including estimated cost and tokens when you provide them.aimake buildexecutes only stale work in topological order (parallel where safe).- Successful outputs land in a content-addressable cache under
.aimake/cache/. aimake explaintells you why a target is stale when something surprises you.
Fingerprints use content hashes. Touching a file's mtime without changing bytes does not invalidate the cache. Details are in How aimake works and Fingerprints & caching.
A concrete example
The sample project at examples/rag/ is a complete RAG pipeline:
cd examples/rag
aimake build # first run: all artifacts execute
aimake build # second run: 0 rebuilt, reused from cache
Edit prompts/system.txt, then:
aimake plan # prompt → evaluation → report marked for rebuild
aimake build # only downstream artifacts run
aimake explain report
aimake diff prompt
Upstream steps (dataset, preprocess, embeddings, index) stay cached because their fingerprints did not change.
Who aimake is for
- Teams that change prompts, models, or configs often and hate rerunning the whole pipeline
- Engineers who want
aimake planto show cost and tokens before spending API budget - Pipelines that are local-first (laptop + CI) with an optional shared S3 cache
- Projects that need quality gates and output validation on evaluation artifacts
If you primarily need Git-linked dataset versioning, keep DVC (aimake has a DVC plugin). If you need cluster-wide scheduling with SLAs, keep Prefect or Airflow and call aimake build as one step.
Next steps
- Install aimake (Python 3.11+)
- Follow the Quick start
- Learn Core concepts — artifacts, fingerprints, plans, locks
- Migrate an existing project with Migration (
--from=makefile|dvc|prefect|airflow-dag)