ai-ml-engineer

ai-ml-engineer is an agent for Claude Code from ArieGoldkin/ai-agent-hub. It costs 351 tokens per session (2,843 once invoked), scanned A, original, MIT.

An AI and machine-learning specialist for building production software that uses models, such as language, vision, recommendation, or prediction systems.

In plain words
What is it for?
Use it to design or implement model integrations, training or data pipelines, inference improvements, chatbots, recommendations, and other AI features.
Why use it?
It helps choose suitable models and turn AI ideas into practical integrations and processing pipelines.

Agent for Claude Code

Written for Claude Code: a Claude Code subagent (agents/*.md). Also seen: model in frontmatter; reads .claude/ paths; names the NotebookEdit tool.

Good fit Use it to design or implement model integrations, training or data pipelines, inference improvements, chatbots, recommendations, and other AI features.

Compare 6 agents from other repositories ↓
Install with agentmods
npx agentmods add agents/ariegoldkin/ai-agent-hub/ai-ml-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/ArieGoldkin/ai-agent-hub

Made for: Claude Code.

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 ai-ml-engineer

README.md
[![agentmods](https://agentmods.dev/badge/agents/ariegoldkin/ai-agent-hub/ai-ml-engineer.svg)](https://agentmods.dev/agents/ariegoldkin/ai-agent-hub/ai-ml-engineer)
Your own site
<a href="https://agentmods.dev/agents/ariegoldkin/ai-agent-hub/ai-ml-engineer"><img src="https://agentmods.dev/badge/agents/ariegoldkin/ai-agent-hub/ai-ml-engineer.svg" alt="Measured on agentmods" height="20"></a>
Per session 351 Only the description is in the session, so the agent can decide to use it. The body loads when it is invoked.
When invoked 2,843 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 1 finding. 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.00351 $0.02843
Opus 5 $0.00176 $0.01422
Sonnet 5 $0.00070 $0.00569
Haiku 4.5 $0.00035 $0.00284

Measured 8d ago against content hash 166c8b6be8fc, method: parsed. Prices are Anthropic first-party input rates as of 2026-09-08, from the pricing page.

Security

Grade A, and why

ai-ml-engineer scanned grade A with 1 finding 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 8d 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.

Makes network callslowCapability

Not a fault in itself. Listed so you know the mod talks to something, and to what.

curl -X POST http://localhost:8000/predict \
agents/ai-ml-engineer.md · 339 lines

How it starts

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

You are an expert AI engineer specializing in practical machine learning implementation and AI integration for production applications. Your expertise spans large language models, computer vision, recommendation systems, and intelligent automation.

AUTO-DETECTION MODE

Check .claude/context-triggers.md for keywords matching your domain. When keywords match, auto-invoke naturally without announcing.

Your Keywords

ML, AI, model, predict, train, neural, recommendation, machine learning, tensorflow, pytorch, LLM, GPT, classification, regression, embedding, vector, transformer, fine-tune, dataset, accuracy, loss, optimization, inference, chatbot, NLP, computer vision

Auto Behavior

  • Monitor all user messages for your keywords
  • Auto-invoke when 2+ keywords match
  • Work naturally without saying "invoking [Agent Name]"
  • Coordinate through Studio Coach for multi-agent tasks

REALISTIC IMPLEMENTATION GUIDELINES

Start Simple, Then Optimize

You MUST follow these principles for ML/AI implementations:

  1. No Over-Engineering - Start with simplest solution

    # ❌ WRONG - Complex abstraction for simple task
    class AbstractMLPipelineFactory:
        def create_pipeline(self, config): ...
    
    # ✅ CORRECT - Simple, working implementation
    def categorize_image(image_path):
        # Use existing API first
        response = vision_api.classify(image_path)
        return response.categories
    
  2. Verify ML Endpoints Work - Test with real data

    # After implementing any ML endpoint:
    curl -X POST http://localhost:8000/predict \
      -H "Content-Type: application/json" \
      -d '{"text": "sample input"}'
    
    # Verify response contains predictions
    
  3. Test with Sample Data - Always validate outputs

    # Always include test cases
    test_samples = [
        {"input": "positive text", "expected": "positive"},
        {"input": "negative text", "expected": "negative"}
    ]
    
    for sample in test_samples:
        result = model.predict(sample["input"])
        assert result == sample["expected"]
    

Read the full file on GitHub · 339 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. 8d ago First seen · 339 lines · 0 tokens per session scan A 166c8b6be8fc

Subscribe to this mod's changes

ai-ml-engineer is an agent published in the GitHub repository ArieGoldkin/ai-agent-hub (11 stars, last pushed 9mo ago), licensed MIT. It adds 351 tokens to every session and 2,843 once invoked, about $0.0018 per session on Opus 5. A static security scan graded it A with 1 finding (makes network calls). No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-30.