Skip to content

Kumo AI (KumoRFM-2)

What it is

Kumo AI is a predictive AI platform that specializes in Relational Foundation Models (RFMs). Its flagship model, KumoRFM-2, is designed to reason over structured, relational data living in enterprise data warehouses. It treats the entire database as a graph, enabling advanced predictive analytics without complex feature engineering. By July 2026, Kumo has integrated support for the MCP 3.0 Task Protocol, allowing autonomous agents to trigger and consume predictions as part of larger automated workflows.

What problem it solves

Traditional machine learning requires data scientists to "flatten" multi-table relational data into a single table (feature engineering), which often destroys valuable predictive signals stored in the relationships between tables. KumoRFM-2 works directly on the graph of connected tables, preserving foreign-key relationships and patterns.

Where it fits in the stack

AI & Knowledge / Process Understanding. It acts as a predictive intelligence layer on top of raw data infrastructure, serving as a "Predictive Engine" for enterprise data.

Typical use cases

  • Zero-Training Predictions: Point the model at a data warehouse and run predictive queries in plain English without task-specific training.
  • Relational Reasoning: Predicting outcomes (e.g., customer churn, product demand) by analyzing patterns across multiple linked tables.
  • Large-Scale Data Science: Scales to over 500 billion rows of relational data, suitable for massive enterprise datasets.
  • Autonomous Decisioning: Using an agent to query Kumo for churn risk and then automatically triggering a retention workflow via n8n.

Strengths

  • No ETL/Feature Engineering: Eliminates the need for complex data pipelines or feature stores.
  • Hierarchical In-Context Learning: Extracts task-aware features at both individual table and cross-table levels.
  • High Performance: Outperforms fully supervised machine learning models on relational benchmarks like RelBench.
  • Predictive Querying: Allows data teams to ask "What will happen?" instead of just "What happened?".
  • Frontier Integration: Optimized for use with Gemma 3, Claude 4.8 Opus, and GPT-5.5 for interpreting predictive results.

Limitations

  • Relational Focus: Primarily designed for structured tabular data, not unstructured text or media.
  • Enterprise Scale: Optimized for large data warehouses (Snowflake, Databricks, BigQuery); may be overkill for simple datasets.
  • Closed Platform: Managed service; not available for local or air-gapped execution.

When to use it

  • When you need to extract predictive insights from complex, multi-table relational databases.
  • To reduce the time-to-value for new data science projects from months to hours.
  • When traditional tabular ML models (XGBoost, etc.) fail to capture signal from relationships.
  • When you want to enable predictive capabilities for autonomous agents via MCP.

When not to use it

  • For tasks involving primarily unstructured data (text, images).
  • For very small or single-table datasets where traditional ML is sufficient.
  • When you require a fully open-source or local predictive stack (consider Ludwig).

Getting started

To get started with Kumo, you connect your cloud data warehouse (Snowflake, Databricks, or BigQuery) and define predictive tasks using a SQL-like interface.

Predictive Querying Examples

-- Predict the total revenue from a customer over the next 90 days
PREDICT SUM(Transactions.Amount)
FOR EACH Users.ID
OVER NEXT 90 DAYS

-- Predict which users will not have any transactions in the next 30 days (Churn)
PREDICT COUNT(Transactions.ID) == 0
FOR EACH Users.ID
OVER NEXT 30 DAYS

CLI examples

[!NOTE] As of July 2026, Kumo AI focuses on its Managed SaaS interface and REST API. There is no official standalone CLI for model management. However, developers can use the Claude Code CLI with the Kumo MCP server to run predictive queries.

API examples

Once a model is trained on Kumo, results can be retrieved via the Kumo REST API or pushed back into your data warehouse.

Prediction Retrieval (Python)

import requests
import os

KUMO_API_KEY = os.environ["KUMO_API_KEY"]
PLAN_ID = "plan_123abc"

def get_predictions(plan_id):
    url = f"https://api.kumo.ai/v1/plans/{plan_id}/predictions"
    headers = {"Authorization": f"Bearer {KUMO_API_KEY}"}

    response = requests.get(url, headers=headers)
    return response.json()

# Fetch latest predictions for high-churn-risk users
predictions = get_predictions(PLAN_ID)
for user in predictions['data']:
    print(f"User: {user['id']}, Churn Probability: {user['score']}")

Sources / references

Contribution Metadata

  • Last reviewed: 2026-07-21
  • Confidence: high