machine-learning-engineer

machine-learning-engineer is an agent for Claude Code from nodnarbnitram/claude-code-extensions. It costs 42 tokens per session (1,407 once invoked), scanned A, original, MIT.

An expert role for deploying machine-learning models, the software systems that make predictions from data, into production services. It covers model optimization, serving infrastructure, scaling, monitoring, and release processes.

In plain words
What is it for?
Designing inference services, improving latency and throughput, setting up CI/CD, containers, registries, autoscaling, monitoring, validation, and progressive rollouts.
Why use it?
It provides a structured way to address the operational work needed to run models reliably for real users, including performance, versioning, and rollback planning.

Agent for Claude Code

Written for Claude Code: shipped in a Claude Code plugin.

Part of the cce-ai plugin — 5 agents shipped together

Good fit Designing inference services, improving latency and throughput, setting up CI/CD, containers, registries, autoscaling, monitoring, validation, and progressive rollouts.

Compare 6 agents from other repositories ↓
Install with agentmods
npx agentmods add agents/nodnarbnitram/claude-code-extensions/machine-learning-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/nodnarbnitram/claude-code-extensions

Made for: Claude Code.

Or install cce-ai, the plugin that ships this one along with the rest of its 5 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 machine-learning-engineer

README.md
[![agentmods](https://agentmods.dev/badge/agents/nodnarbnitram/claude-code-extensions/machine-learning-engineer/github.svg)](https://agentmods.dev/agents/nodnarbnitram/claude-code-extensions/machine-learning-engineer)
Your own site
<a href="https://agentmods.dev/agents/nodnarbnitram/claude-code-extensions/machine-learning-engineer"><img src="https://agentmods.dev/badge/agents/nodnarbnitram/claude-code-extensions/machine-learning-engineer/github.svg" alt="Measured on agentmods" height="20"></a>

Or the 80×15 button, for a site that already has a row of RSS and ATOM ones. Only the verdict fits; the numbers stay here.

agentmods 80×15 button for machine-learning-engineer

Your own site · 80×15
<a href="https://agentmods.dev/agents/nodnarbnitram/claude-code-extensions/machine-learning-engineer"><img src="https://agentmods.dev/badge/agents/nodnarbnitram/claude-code-extensions/machine-learning-engineer.svg" alt="Reviewed on agentmods" width="80" height="20"></a>
Per session 42 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,407 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.00042 $0.01407
Opus 5.5 $0.00017 $0.00563
Sonnet 5.5 $0.00008 $0.00281
Haiku 4.5 $0.00004 $0.00141

Measured 1mo ago against content hash d07c647b63c7, method: parsed. Prices are Anthropic first-party input rates as of 2026-10-07, from the pricing page.

Security

Grade A, and why

machine-learning-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 1mo 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.

Origin

Copies of this mod

1 near-identical copy found in the catalogue:

plugins/cce-ai/agents/machine-learning-engineer.md · 286 lines

How it starts

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

You are a senior machine learning engineer with deep expertise in deploying and serving ML models at scale. Your focus spans model optimization, inference infrastructure, real-time serving, and edge deployment with emphasis on building reliable, performant ML systems that handle production workloads efficiently.

When invoked:

  1. Query context manager for ML models and deployment requirements
  2. Review existing model architecture, performance metrics, and constraints
  3. Analyze infrastructure, scaling needs, and latency requirements
  4. Implement solutions ensuring optimal performance and reliability

ML engineering checklist:

  • Inference latency < 100ms achieved
  • Throughput > 1000 RPS supported
  • Model size optimized for deployment
  • GPU utilization > 80%
  • Auto-scaling configured
  • Monitoring comprehensive
  • Versioning implemented
  • Rollback procedures ready

Model deployment pipelines:

  • CI/CD integration
  • Automated testing
  • Model validation
  • Performance benchmarking
  • Security scanning
  • Container building
  • Registry management
  • Progressive rollout

Serving infrastructure:

  • Load balancer setup
  • Request routing
  • Model caching
  • Connection pooling
  • Health checking
  • Graceful shutdown
  • Resource allocation
  • Multi-region deployment

Model optimization:

  • Quantization strategies
  • Pruning techniques
  • Knowledge distillation
  • ONNX conversion
  • TensorRT optimization
  • Graph optimization
  • Operator fusion
  • Memory optimization

Batch prediction systems:

  • Job scheduling
  • Data partitioning
  • Parallel processing
  • Progress tracking
  • Error handling
  • Result aggregation
  • Cost optimization
  • Resource management

Real-time inference:

  • Request preprocessing
  • Model prediction
  • Response formatting
  • Error handling
  • Timeout management
  • Circuit breaking
  • Request batching
  • Response caching

Performance tuning:

  • Profiling analysis
  • Bottleneck identification
  • Latency optimization
  • Throughput maximization
  • Memory management
  • GPU optimization
  • CPU utilization
  • Network optimization

Auto-scaling strategies:

  • Metric selection
  • Threshold tuning
  • Scale-up policies
  • Scale-down rules
  • Warm-up periods
  • Cost controls
  • Regional distribution
  • Traffic prediction

Multi-model serving:

  • Model routing
  • Version management
  • A/B testing setup
  • Traffic splitting
  • Ensemble serving
  • Model cascading
  • Fallback strategies
  • Performance isolation

Edge deployment:

  • Model compression
  • Hardware optimization
  • Power efficiency
  • Offline capability
  • Update mechanisms
  • Telemetry collection
  • Security hardening
  • Resource constraints

MCP Tool Suite

  • tensorflow: TensorFlow model optimization and serving
  • pytorch: PyTorch model deployment and optimization
  • onnx: Cross-framework model conversion
  • triton: NVIDIA inference server
  • bentoml: ML model serving framework
  • ray: Distributed computing for ML
  • vllm: High-performance LLM serving

Read the full file on GitHub · 286 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. 1mo ago First seen · 286 lines · 42 tokens per session scan A d07c647b63c7

Subscribe to this mod's changes

machine-learning-engineer is an agent published in the GitHub repository nodnarbnitram/claude-code-extensions (16 stars, last pushed 6mo ago), licensed MIT. It adds 42 tokens to every session and 1,407 once invoked, about $0.0002 per session on Opus 5.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-30.

Related

Other agents, from other repositories

data-engineer

Build scalable data pipelines, modern data warehouses, and real-time streaming architectures. Implements Apache Spark, dbt, Airflow, and cloud-native data platforms. Use PROACTIVELY for data pipeline design, analytics infrastructure, or modern data stack implementation.

wshobson/agents · 54 tokens

FabricDataEngineer

Orchestrate end-to-end Microsoft Fabric data engineering workflows that span multiple workloads and personas. Use when the request crosses Spark, Warehouse, Pipelines, Lakehouse architecture, migration, or data quality operations. Delegates deep single-endpoint implementation to specialized skills and resources.

microsoft/skills-for-fabric · 58 tokens

mlops_engineer

MLOps specialist for model registry, CI/CD for models, deployment, monitoring, and drift detection. Use when the task requires packaging models for serving, building training/deploy pipelines, configuring model monitoring, or wiring up canary rollouts. For example: automating retraining on a schedule, setting up…

josstei/maestro-orchestrate · 76 tokens

dbt-deployer

Deploys local dbt projects to dbt Projects on Snowflake and manages CI/CD pipelines.

sfc-gh-dflippo/snowflake-dbt-demo · 23 tokens

FAI Azure AKS Expert

Azure Kubernetes Service specialist — GPU node pools (A100/H100), NVIDIA device plugin, model serving with vLLM/TGI/Triton, HPA/KEDA autoscaling, and production AI inference workload patterns.

frootai/frootai · 51 tokens

FAI Azure Storage Expert

Azure Storage specialist — Blob lifecycle tiers, ADLS Gen2 for data lakes, private endpoints, managed identity auth, and document/model artifact storage for AI pipelines.

frootai/frootai · 38 tokens