staff-ml-engineer

Guide end-to-end ML workflows with scikit-learn, MLflow, Optuna, and FastAPI.

1|Updated Apr 27, 2026
One-click install
npx skills add https://github.com/WayneBanksy/wayneys_claude --skill staff-ml-engineer
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: staff-ml-engineer
Source: https://github.com/WayneBanksy/wayneys_claude/tree/main/skills/staff-ml-engineer
Command: npx skills add https://github.com/WayneBanksy/wayneys_claude --skill staff-ml-engineer

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires mlflow, scikit-learn, xgboost, prophet, pytorch, optuna, evidently, and includes scripts (resource) and references (resource) components.

What problem does it solve?

This Skill facilitates comprehensive machine learning development, tracking, and deployment, reducing complexity and errors in production ML systems.

Core Features & Use Cases

  • Model Development: Builds robust pipelines for training classical ML, time-series forecasting, and deep learning models.
  • Experiment Tracking: Automates MLflow experiments for reproducibility and auditability.
  • Hyperparameter Optimization: Integrates with Optuna for efficient, automated tuning.
  • Model Serving & Monitoring: Implements FastAPI endpoints for deployment and Evidently reports for drift detection.
  • Use Case: Example—training and deploying a time-series forecast with Prophet, tracking parameters and metrics, and monitoring data drift in production.

Quick Start

Create a ML pipeline that trains a classification model, logs parameters and metrics with MLflow, and serve it via FastAPI.

Frequently Asked Questions about staff-ml-engineer

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I track machine learning experiments and reproduce model training metrics?

Track machine learning experiments by automating MLflow experiment logging for parameters and metrics, ensuring reproducibility and auditability across classical ML, time-series, and deep learning models throughout the training pipeline.

What is the best way to serve a trained machine learning model via an API endpoint?

Serve a trained machine learning model by implementing FastAPI endpoints for deployment, enabling production-ready model serving directly from your tracked scikit-learn, PyTorch, or Prophet forecasting pipelines.

Can I use Optuna for hyperparameter tuning with scikit-learn and PyTorch models?

Optuna supports efficient, automated hyperparameter tuning for both scikit-learn and PyTorch models, integrating directly into the machine learning training workflow to optimize model parameters systematically.

How do I monitor data drift in production machine learning systems?

Monitor data drift in production machine learning systems by generating Evidently reports, detecting distribution shifts and maintaining safety in deployed models served via FastAPI endpoints.

Does this machine learning workflow support time-series forecasting with Prophet?

The machine learning workflow explicitly supports time-series forecasting with Prophet, allowing you to train forecasts, log parameters with MLflow, and deploy the model via FastAPI endpoints.

What are the limitations of automating end-to-end machine learning deployment pipelines?

End-to-end machine learning deployment pipelines require managing dependencies across MLflow, scikit-learn, PyTorch, and Evidently, demanding careful environment configuration to ensure reproducibility and avoid production drift issues.