How Feature Stores Works Under the Hood
TL;DR
Here is a clear, practical guide to under the hood: the fundamentals, the best practices that actually move the needle, common mistakes to avoid, concrete data points, and a short FAQ. Everything is structured so you can apply it to real projects today.
Key takeaways
- A model registry (MLflow, Unity Catalog, SageMaker) is the single source of truth for what is deployed, its lineage, and its promotion stage, so wire it into your CI/CD before you scale.
- A feature store solves training-serving skew by computing features once and serving the identical logic to both offline training and online inference paths.
- Monitor inputs and predictions in production for drift, not just uptime, because a silently degrading model fails the business long before it throws an error.
- Evaluate LLM applications with a versioned test set and a mix of deterministic checks and LLM-as-judge scoring, and gate deployments on those evals in CI.
- Put an AI gateway (LiteLLM, Portkey, Cloudflare AI Gateway) in front of your LLM calls to centralize keys, rate limits, caching, fallbacks, and cost tracking across providers.
This is a practical, up-to-date guide to Under the Hood — what it is, why it matters in 2026, and how to apply it in real projects. It is written for developers and founders who want clear answers and proven best practices, not filler.
Whether you're just starting out or leveling up, treat this as a working reference you can return to. Every section is built to be skimmed, applied, and shared.
Model serving with vLLM and TGI
Model serving is the runtime layer that turns a trained model into a low-latency, high-throughput API, and for open-weight LLMs the dominant engines are vLLM and Hugging Face Text Generation Inference. vLLM introduced PagedAttention, which manages the attention key-value cache in non-contiguous pages so that GPU memory is used efficiently and many requests can be batched together, while TGI offers a production-hardened server with tensor parallelism, quantization, and streaming. Both rely on continuous (in-flight) batching, where new requests join a running batch instead of waiting for a fixed window, which is the single biggest lever for GPU utilization. Alternatives and complements include NVIDIA Triton with its TensorRT-LLM backend, SGLang, and managed endpoints, but vLLM has become the common default for self-hosting.
Prompt management and versioning
As prompts become load-bearing logic, teams need to manage them like code rather than scattering string literals across a codebase. Prompt management systems store prompts as versioned, named templates with variables, track which version is deployed, and link each version to its evaluation results so changes are measurable rather than vibes-based. This lets non-engineers iterate on prompts in a UI while engineers keep production changes gated behind review and evals, and it enables A/B testing and instant rollback of a bad prompt. Platforms such as LangSmith, Langfuse, PromptLayer, Humanloop, and Braintrust provide prompt registries, playgrounds, and linkage to traces. The core principle is that a prompt is a deployable artifact with a lifecycle, not an incidental string.
CI/CD for machine learning
CI/CD for ML extends the familiar build-test-deploy pipeline to cover data and models, which introduces stages that software pipelines do not have. Beyond running unit tests on code, an ML pipeline validates incoming data schemas and quality, triggers training when new data or code arrives, evaluates the resulting model against a holdout set and the current production model, and only promotes it if it clears the bar. Continuous training, where retraining is automated on a schedule or triggered by drift alerts, is the ML-specific addition that keeps models fresh. Orchestrators such as Kubeflow Pipelines, Metaflow, Airflow, Dagster, and ZenML define these workflows as code, while DVC and Git-based data versioning make each run reproducible from data to model.
AI gateways as a control plane
An AI gateway is a proxy that sits between your applications and one or more model providers, giving you a single control point for reliability, cost, and governance. Instead of every service holding its own API keys and retry logic, calls route through the gateway, which handles authentication, rate limiting, retries, provider fallback, load balancing, and semantic caching to avoid paying for repeated identical calls. Gateways also centralize observability and spend tracking, tagging usage by team or feature so finance can attribute cost, and they enforce guardrails and PII redaction in one place. Popular options include LiteLLM, Portkey, Cloudflare AI Gateway, Kong AI Gateway, and cloud-native offerings, and many expose an OpenAI-compatible interface so switching backends requires no application changes.
Common pitfalls and how to avoid them
The most common failure in ML systems is training-serving skew, where offline and online feature computation quietly diverge, which is best prevented with a shared feature-serving path or feature store. A close second is shipping without production monitoring, so a model degrades from drift for weeks before anyone notices, which argues for wiring drift and prediction monitoring in from day one. Teams also over-engineer early, adopting a heavy platform before they have a single model in production, when a simpler stack of MLflow plus a scheduler would have shipped faster. For LLM applications, the recurring traps are treating evaluation as an afterthought, hardcoding prompts and keys instead of centralizing them behind a registry and gateway, and underestimating token cost until the bill arrives; each is avoidable by building evals, versioning, and a gateway in early.
How LLMOps differs from classic MLOps
LLMOps is the specialization of MLOps for applications built on large language models, and it shifts the center of gravity from training your own models to orchestrating, prompting, and evaluating foundation models you often did not train. Classic MLOps assumes you own the training pipeline and can retrain to fix drift; with hosted LLMs you instead manage prompts, retrieval pipelines, tool definitions, and provider selection. Evaluation becomes harder because outputs are open-ended and non-deterministic, pushing teams toward LLM-as-judge scoring and human review rather than a single accuracy number. New operational primitives appear too, such as token-cost budgeting, prompt versioning, semantic caching, and guardrails against prompt injection and unsafe output.
Under the Hood: Key Facts and Data
According to recent industry research and the official documentation linked below:
- Kubernetes has become the de facto substrate for GPU orchestration in production ML, with the NVIDIA device plugin, GPU Operator, and schedulers such as Kueue, Volcano, and Run:ai handling accelerator allocation.
- MLflow, open-sourced by Databricks in 2018, has become one of the most popular experiment-tracking and model-registry tools, reporting tens of millions of monthly downloads by the mid-2020s.
- Industry surveys have repeatedly indicated that a large majority of ML projects never reach production, with figures often cited in the range of 70-90 percent, a gap that MLOps tooling is explicitly designed to close.
Quick-Reference Summary
A map of what this guide covers:
| Topic | What you'll learn |
|---|---|
| Model serving with vLLM and TGI | Model serving is the runtime layer that turns a trained model into a low-latency |
| Prompt management and versioning | As prompts become load-bearing logic, teams need to manage them like code rather than scattering string literals across |
| CI/CD for machine learning | CI/CD for ML extends the familiar build-test-deploy pipeline to cover data and models |
| AI gateways as a control plane | An AI gateway is a proxy that sits between your applications and one or more model providers |
| Common pitfalls and how to avoid them | The most common failure in ML systems is training-serving skew |
| How LLMOps differs from classic MLOps | LLMOps is the specialization of MLOps for applications built on large language models |
How to Get Started with Under the Hood
A simple path that works:
- Learn the fundamentals of Under the Hood from primary sources, not just tutorials.
- Build one small, real project end to end.
- Get feedback, refactor, and add tests.
- Ship it publicly and document what you learned.
- Repeat with a slightly harder project each time.
Build It with a World-Class Full Stack Developer
Sandeep Kumar Chaudhary is a full stack world-class developer. If you want to turn this into a real, production-ready product, get in touch — message directly on WhatsApp at +9779802348957 for a fast, no-pressure consult.
You can also explore the projects already shipped to thousands of users, or start a conversation here.
Final Thoughts
A model registry (MLflow, Unity Catalog, SageMaker) is the single source of truth for what is deployed, its lineage, and its promotion stage, so wire it into your CI/CD before you scale. The developers and teams who win in 2026 pair strong fundamentals with consistent shipping. Start small, stay curious, build in public, and revisit this guide as your skills grow.
Sources and Further Reading
Frequently Asked Questions
What is under the hood?
As prompts become load-bearing logic, teams need to manage them like code rather than scattering string literals across a codebase. Prompt management systems store prompts as versioned, named templates with variables, track which version is deployed, and link each version to its evaluation results so changes are measurable rather than vibes-based. This guide covers under the hood end to end — core concepts, best practices, concrete data, and a step-by-step approach you can apply right away.
What does a model registry do?
A model registry is the source of truth for trained models: it stores each version with its metrics, parameters, and lineage back to the data and code that produced it, and it manages promotion stages like staging and production with approval workflows. Deployment tooling reads from it to know exactly which version to serve, and it makes rollbacks and audits straightforward. MLflow Model Registry is the common open-source choice, with SageMaker, Vertex AI, and Databricks Unity Catalog offering equivalents.
vLLM or TGI for serving open-source LLMs?
Both are strong, production-grade inference engines built around continuous batching. vLLM is known for its PagedAttention memory management and broad model and quantization support and has become the common open-source default, while Hugging Face TGI integrates tightly with the Hugging Face ecosystem and is battle-tested in their inference stack. Benchmark both on your specific model, hardware, and traffic pattern, since results vary; NVIDIA Triton with TensorRT-LLM is worth testing when you need maximum optimization on NVIDIA hardware.
What is LLMOps and is it just MLOps rebranded?
LLMOps is MLOps specialized for applications built on large language models, and it is more than a rebrand because the operational primitives genuinely differ. You typically orchestrate hosted foundation models rather than training your own, so the work centers on prompt versioning, retrieval pipelines, non-deterministic evaluation, token-cost management, and safety guardrails rather than retraining loops. The underlying discipline of versioning, testing, monitoring, and CI/CD carries over, but the specific tools and failure modes are distinct.
What is the difference between MLOps and DevOps?
DevOps automates building, testing, and deploying software whose behavior is fully determined by its code. MLOps adds the data and model dimension: it versions datasets, tracks experiments, manages a model registry, and monitors for drift, because an ML system's behavior depends on data that changes over time. In short, MLOps is DevOps plus continuous training and continuous monitoring of models.
Sandeep Kumar Chaudhary
Full Stack Software Developer· Nepal's SEO, AEO, GEO & AIO expert and share-market educator. More about me
