CloudSketch AI Logo
FeaturesTemplatesPricingEnterpriseAboutContactLog in
🌙☀️
Log inStart free
Home/Templates/AI & Machine Learning

🧠 AI & Machine Learning · Deployment · Pro

MLOps Model Training Pipeline

Where each part of an MLOps setup runs: feature store, training jobs on GPU nodes, experiment tracking, model registry and automated deployment to serving.

More AI & Machine Learning templates

Deployment diagrams are part of Pro. Anyone can view this one; generating and editing it needs Pro.

Drawing diagram…

What this diagram shows

  • Training runs on a separate GPU node pool
  • Every model version is registered with its metrics
  • Only approved models are promoted to production

Prompt used

Deployment of an MLOps platform on Google Cloud: data lands in BigQuery and a Feast feature store. Kubeflow Pipelines on GKE runs training jobs on a GPU node pool, logs runs to MLflow, and registers models in the MLflow model registry. After approval, Argo CD deploys the model to a KServe serving cluster behind a load balancer. Models and artifacts are stored in Cloud Storage.

Mermaid code
flowchart TB
  DS[Data Scientists] --> KF
  subgraph GCP[Google Cloud]
    BQ[(BigQuery)]
    FS[(Feast Feature Store)]
    GCS[(Cloud Storage - artifacts)]
    subgraph TRAIN[GKE Training Cluster]
      KF[Kubeflow Pipelines]
      subgraph GPU[GPU Node Pool]
        JOB[Training Jobs]
      end
      MLF[MLflow Tracking and Registry]
    end
    subgraph SERVE[GKE Serving Cluster]
      ARGO[Argo CD]
      KS[KServe Model Pods]
    end
    LB[Load Balancer]
  end
  BQ --> FS
  KF --> JOB
  FS --> JOB
  JOB --> MLF
  JOB --> GCS
  MLF -->|Approved version| ARGO
  ARGO --> KS
  GCS --> KS
  LB --> KS
  APPS[Client Apps] --> LB

Related templates

C4 ArchitectureProAI & Machine Learning

RAG Chatbot over Company Documents

A chatbot that answers staff questions from company documents: documents are split and indexed in a vector database, and each question pulls the most relevant passages before the language model writes an answer.

SequenceAI & Machine Learning

RAG Question Answering Sequence

What happens when a user asks the document chatbot a question: permission check, embedding, vector search, prompt building and a cited answer.

Data FlowAI & Machine Learning

Document Ingestion for Vector Search

How documents become searchable passages for an AI assistant: extraction, cleaning, chunking, embedding and indexing, with changed files re-processed automatically.

FlowchartAI & Machine Learning

AI Agent Tool-Use Workflow

How an AI agent completes a task by planning, calling tools, checking results and asking a person to approve risky actions before it finishes.

State MachineAI & Machine Learning

ML Model Lifecycle

The states a machine learning model moves through, from experiment to production to retirement, including rollback when live performance drops.

C4 ArchitectureProAI & Machine Learning

LLM Gateway with Cost and Safety Controls

One gateway that every app in a company uses to reach language models: it applies budgets, rate limits, content filters and caching, and records usage per team.