aiml-engineer

aiml-engineer is an agent for Claude Code from ShaheerKhawaja/ProductionOS. It costs 71 tokens per session (1,326 once invoked), scanned A, original, MIT.

An AI and machine-learning engineering specialist that plans model pipelines, chooses where models run, and sets up model training and evaluation. It covers language, vision, audio, and multimodal models.

In plain words
What is it for?
Use it to design inference or fine-tuning workflows, prepare training data, choose model infrastructure, build evaluation processes, and improve cost or latency.
Why use it?
It helps turn a model idea into a workable system by guiding choices about training, serving, quality checks, cost, and response speed.

Agent for Claude Code

Written for Claude Code: shipped in a Claude Code plugin. Also seen: model in frontmatter.

Part of the productionos plugin — 4 skills, 41 commands, 11 agents shipped together

Good fit Use it to design inference or fine-tuning workflows, prepare training data, choose model infrastructure, build evaluation processes, and improve cost or latency.

Compare 6 agents from other repositories ↓
Install with agentmods
npx agentmods add agents/shaheerkhawaja/productionos/aiml-engineer
Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

Clone the repo
git clone --depth 1 https://github.com/ShaheerKhawaja/ProductionOS

Made for: Claude Code.

Or install productionos, the plugin that ships this one along with the rest of its 4 skills, 41 commands, 11 agents.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

agentmods badge for aiml-engineer

README.md
[![agentmods](https://agentmods.dev/badge/agents/shaheerkhawaja/productionos/aiml-engineer.svg)](https://agentmods.dev/agents/shaheerkhawaja/productionos/aiml-engineer)
Your own site
<a href="https://agentmods.dev/agents/shaheerkhawaja/productionos/aiml-engineer"><img src="https://agentmods.dev/badge/agents/shaheerkhawaja/productionos/aiml-engineer.svg" alt="Measured on agentmods" height="20"></a>
Per session 71 Only the description is in the session, so the agent can decide to use it. The body loads when it is invoked.
When invoked 1,326 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. A grade says what 26 rules found in the file — not that it is safe.
Origin original No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5.1 $0.00071 $0.01326
Opus 5 $0.00036 $0.00663
Sonnet 5 $0.00014 $0.00265
Haiku 4.5 $0.00007 $0.00133

Measured 7d ago against content hash 24391322eef3, method: parsed. Prices are Anthropic first-party input rates as of 2026-09-07, from the pricing page.

Security

Grade A, and why

aiml-engineer scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 7d ago.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

agents/aiml-engineer.md · 125 lines

How it starts

The opening of the file, as written. The whole thing — 125 lines — stays where its author put it; the contents beside it link to each section on GitHub.

ProductionOS AI/ML Engineer

Capabilities

1. Model Pipeline Design

  • Select base model for the use case (LLM, vision, audio, multimodal)
  • Design inference pipeline (direct API, self-hosted, edge deployment)
  • Design fine-tuning pipeline (LoRA adapters, dataset preparation, evaluation)
  • Implement prompt engineering (system prompts, few-shot, chain-of-thought)
  • Design evaluation framework (automated evals, human-in-the-loop, A/B testing)

2. Infrastructure Selection

Need Recommended When
Managed inference Replicate, Modal, RunPod Quick deployment, variable load
Self-hosted inference vLLM + GPU instance High volume, cost optimization
Fine-tuning Hugging Face TRL, Axolotl Custom model adaptation
LoRA training Unsloth, PEFT Parameter-efficient fine-tuning
Embeddings Sentence-Transformers, OpenAI RAG pipelines, semantic search
Model serving TGI, vLLM, Triton Production inference
Evaluation LangSmith, Braintrust, custom Quality monitoring

3. LoRA Fine-Tuning Protocol

When the target needs a fine-tuned model:

  1. Data preparation: Format training data (instruction/response pairs, DPO pairs)
  2. Base model selection: Choose smallest model that meets quality bar
  3. LoRA config: rank (8-64), alpha (16-128), target modules
  4. Training: Epochs, learning rate, batch size, gradient accumulation
  5. Evaluation: Benchmark against base model on task-specific metrics
  6. Deployment: Merge adapter or serve with LoRA hot-swapping

4. Cost Optimization

  • Model routing (cheap model for easy tasks, expensive for hard)
  • Prompt caching (save 50-90% on repeated prefixes)
  • Batch inference (throughput optimization)
  • Quantization (4-bit, 8-bit for self-hosted)
  • Semantic caching (cache similar queries, not just exact matches)

Read the full file on GitHub · 125 lines

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 7d ago First seen · 125 lines · 71 tokens per session scan A 24391322eef3

Subscribe to this mod's changes

aiml-engineer is an agent published in the GitHub repository ShaheerKhawaja/ProductionOS (8 stars, last pushed 4mo ago), licensed MIT. It adds 71 tokens to every session and 1,326 once invoked, about $0.0004 per session on Opus 5. A static security scan graded it A with 0 findings. No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-31.

Related

Other agents, from other repositories

ai-eval-engineer

Builds and maintains the eval pipeline for ai-system / agent-product archetypes. Outputs tests/eval/EVAL-.md files (golden citation, refuse-when-uncertain, output schema, prompt injection, cost-overrun, cross-user isolation). Runs regression on every prompt or model change. Detects drift.

avelikiy/great_cto · 71 tokens

data-platform-reviewer

Data-platform pre-implementation reviewer. Outputs threat model TM-{slug}.md and signs off retention + lineage decisions before senior-dev claims tasks.

avelikiy/great_cto · 33 tokens

mlops-reviewer

MLOps / model lifecycle pre-implementation reviewer. Outputs threat model TM-{slug}.md and signs off training-pipeline + serving-strategy decisions before senior-dev claims tasks.

avelikiy/great_cto · 41 tokens

auth-engineer

Authentication and access-control specialist for SMB Product-Builder products. Owns the auth contract — provider choice (Auth.js default / Clerk fast-path), session model, RBAC, multi-tenant row-level isolation, the protected-route map, account lifecycle (signup/login/reset/invite), and OAuth/magic-link/password…

avelikiy/great_cto · 86 tokens

geo-routing-engineer

Geospatial and routing specialist for Product-Builder products with maps, scheduling-by-location, or vehicle routing (route-optimization in logistics, dispatch in home services, field-booking). Owns the routing contract — geocoding, the VRP/routing model (constraints, objective), maps/distance-matrix provider…

avelikiy/great_cto · 112 tokens

us-ai-reviewer

US AI-governance pre-implementation reviewer — the US analogue of the EU AI Act coverage. Outputs threat model TM-usai-{slug}.md and signs off the AI-governance gate before senior-dev claims tasks.

avelikiy/great_cto · 50 tokens