> This page is for Kumo Engine. > For clean Markdown of any page, append .md to the page URL. > For a complete documentation index, see https://docs.nvidia.com/sdgm/llms.txt. > For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://docs.nvidia.com/sdgm/_mcp/server. # Get Started with Agents > Install the Kumo Engine agent skill for Claude Code, Codex, Snowflake Cortex Code, or Databricks Genie Code, then train your first Kumo Engine model from a plain-language request. The fastest way to train a model with Kumo Engine is to describe your prediction problem to the coding agent you already use. The `kumo-data-science` skill teaches the agent the Kumo Engine Python API. You name your data, the outcome to predict, and the time window. The agent inspects your tables, confirms the graph and the task with you, and writes one notebook that trains a task-specific Kumo Engine model on your data and makes predictions. This workflow trains your own model. To make predictions with a pretrained model and no training, use [Structured Data Models](/sdgm/structured-data-models/overview) instead. Without the skill, an agent answers from general knowledge and usually gets the Kumo Engine API wrong. ## Fastest Way to Get Started ### Requirements You need three things: * **A supported agent:** Claude Code, Codex, Snowflake Cortex Code, or Databricks Genie Code. Other agents that read skill directories can use the generic package. * **An environment that runs Kumo Engine:** a Linux machine, a Mac with Apple silicon, Windows through WSL2, a Snowflake notebook on GPU Container Runtime, or a Databricks GPU notebook. The agent writes the notebook. Your environment runs it. Check [Kumo Engine Prerequisites](/sdgm/kumo-engine/get-started-with-kumo-engine/prerequisites) for supported runtimes and [GPU Sizing](/sdgm/kumo-engine/get-started-with-kumo-engine/gpu-sizing) for capacity. * **Read access to your data:** a local folder of CSV or Parquet files, Amazon S3, Google Cloud Storage, BigQuery, Snowflake, or Databricks. The agent reads table names, columns, and types before it writes code. It never changes your source data. You install the skill from a download attached to the `kumo-engine` release, one package per agent host. The skill runs wherever your agent runs, including a Mac. Only the notebooks it writes need an environment that runs Kumo Engine. ### Get Started with Cortex Code or Genie Code Use this path when your data lives in Snowflake or Databricks and you want the agent inside that platform. #### Snowflake Cortex Code **Snowsight workspace.** Download the Cortex Code package and unzip it. Open the workspace where you will do your Kumo work, open Cortex Code settings, choose **Skills**, then **Upload Skill Folder(s)**, and select the `kumo-data-science` folder. The skill is visible to Cortex Code in that workspace only. Download CoCo skill Direct link: **Cortex Code CLI or desktop app.** Install the skill into your user skills directory, `~/.snowflake/cortex/skills/`: ```bash mkdir -p ~/.snowflake/cortex/skills curl -fsSL https://github.com/NVIDIA/kumo-engine/releases/download/kumo-engine-v1.0.1/kumo-data-science-coco.zip -o /tmp/kumo-data-science-coco.zip unzip -oq /tmp/kumo-data-science-coco.zip -d ~/.snowflake/cortex/skills ``` If Claude Code is also installed on the same machine, Cortex Code loads the copy in `~/.claude/skills/` first. To give Cortex Code its own copy, unzip into your project's `./.cortex/skills/` instead. #### Databricks Genie Code Download the Genie Code package and unzip it. In your Databricks workspace, upload the `kumo-data-science` folder to `/Users//.assistant/skills/`. Start a new Genie Code chat to load the skill. Download Genie skill Direct link: ### Get Started with Claude Code or Codex Use this path when your agent runs in a terminal or editor beside you. The agent writes code for any of the six Kumo connectors, so your data can be local or in a warehouse. Paste these commands into a terminal to install the skill. They work on Linux and macOS and need no Python package: #### Claude Code ```bash mkdir -p ~/.claude/skills curl -fsSL https://github.com/NVIDIA/kumo-engine/releases/download/kumo-engine-v1.0.1/kumo-data-science-generic.zip -o /tmp/kumo-data-science.zip unzip -oq /tmp/kumo-data-science.zip -d ~/.claude/skills ``` #### Codex ```bash mkdir -p ~/.agents/skills curl -fsSL https://github.com/NVIDIA/kumo-engine/releases/download/kumo-engine-v1.0.1/kumo-data-science-generic.zip -o /tmp/kumo-data-science.zip unzip -oq /tmp/kumo-data-science.zip -d ~/.agents/skills ``` Run them again to upgrade. Start a new agent session to load the skill. | Agent | User skills directory | Project skills directory | | ---------------------------- | --------------------- | ------------------------ | | Claude Code | `~/.claude/skills/` | `./.claude/skills/` | | Codex | `~/.agents/skills/` | `./.agents/skills/` | | Genie Code CLI (`genie-cli`) | `~/.agents/skills/` | `./.agents/skills/` | To install by hand, download the generic skill and unzip it into a directory in the table above. On a Mac with Apple silicon, follow [macOS Setup](/sdgm/kumo-engine/get-started-with-kumo-engine/installation/macos) to run locally. On Windows, install the skill and run your workflow inside [WSL2](/sdgm/kumo-engine/get-started-with-kumo-engine/installation/windows-wsl2). The same download works for any other agent that reads skill directories. Download skill Direct link: ### Check the Skill Is Loaded Start a new chat and ask: ```text Load the kumo-data-science skill and quote its SKILL-BUILD line. ``` The agent quotes a `SKILL-BUILD:` line from the skill. A vague answer or an invented version means the agent did not load the skill. Check the install location and start a new chat. After you upgrade `kumo-engine`, download the skill from the matching release and install it again. ## Quick Example: Predict Repeat Customers This example uses AdventureWorks, Microsoft's public sample of a bicycle retailer: customers, orders, order lines, and products. You ask the agent to train a Kumo Engine model that predicts which customers will order again in the next 90 days. #### Get the data Download the [Linux examples bundle](/sdgm/kumo-engine/examples/linux-local-examples#prepare-the-input-files) and run its two preparation scripts. They download AdventureWorks from Microsoft, verify the checksums, and write four CSV files. For Cortex Code or Genie Code, load the four CSV files as tables in a schema you can read, for example `KUMO_DEMO.ADVENTUREWORKS`. #### Ask your agent Paste one of these prompts into a new chat. Each one names the data, the entity, the outcome, and the window, which is everything the skill needs. #### Claude Code or Codex ```text Train a Kumo model on the AdventureWorks CSV files in ./adventureworks-inputs. For each customer, predict whether they place at least one order in the next 90 days. ``` #### Cortex Code or Genie Code ```text Train a Kumo model on the tables in KUMO_DEMO.ADVENTUREWORKS. For each customer, predict whether they place at least one order in the next 90 days. ``` #### Confirm the graph The agent reads the tables and replies with the primary keys, time columns, and links it inferred, each marked as inferred. Correct any line, or reply `Confirmed`. #### Confirm the task The agent restates the task: the entity (customers), the outcome (at least one order), the window (90 days), and the task type (binary classification). Reply `Confirmed`. #### Run the notebook The agent writes one notebook that builds the graph, trains the model, and predicts for every customer. It does not run anything until you ask. Ask it to run the notebook, or open it and run it yourself. You get evaluation metrics for the trained model and a probability for each customer. To compare with a hand written version of the same task, see [Predict a Yes or No Outcome on Linux](/sdgm/kumo-engine/user-code-guide/code-examples/linux-deployment/binary-classification). After the first model, keep going in the same chat. Ask the agent to explain the predictions, diagnose a training run, tune the model, or save it and score new data later. ## For AI Agents These docs publish a machine-readable index for agents, in the same spirit as `robots.txt`: * [`llms.txt`](https://docs.nvidia.com/sdgm/llms.txt) lists every page with a short summary. * [`llms-full.txt`](https://docs.nvidia.com/sdgm/llms-full.txt) contains the full text of the docs. To set up an agent that does not have the skill yet, paste this into its chat: ```text Install the Kumo Engine agent skill for yourself. Download https://github.com/NVIDIA/kumo-engine/releases/download/kumo-engine-v1.0.1/kumo-data-science-generic.zip and unzip it into your skills directory. Use https://docs.nvidia.com/sdgm/llms.txt as the reference for Kumo Engine. Then tell me the SKILL-BUILD line of the kumo-data-science skill. ``` ## Next Steps #### [Write the Request Yourself](/sdgm/kumo-engine/user-code-guide/train-a-model/predictive-queries) Learn how predictive queries express the entity, outcome, and window your agent asks for. #### [Set Up Your Kumo Environment](/sdgm/kumo-engine/get-started-with-kumo-engine/available-deployment-options) Choose a Linux, Snowflake, or Databricks deployment to run the notebooks your agent writes.