Kumo · 1 month ago
AI Engineer - Relational Foundation Models & Agentic Systems
Mountain View, CA
Full-time
Onsite
Entry, Mid Level
$125K/yr - $185K/yr
1+ years expKumo is redefining enterprise AI with foundation models for relational data, enabling organizations to predict, optimize, and act with speed and confidence. The role involves designing and building AI-powered products, collaborating with cross-functional teams, and directly interfacing with customers to shape the growth of Kumo's Relational Foundation Model.
Artificial Intelligence (AI)Big DataBusiness IntelligenceMachine LearningSaaS
Responsibilities
Design, implement, and deploy AI agents that assist data scientists on relational/SQL data and recommend next-best actions
Build user-centric APIs and product surfaces (web/UI or programmatic) that make agentic workflows feel seamless and reliable
Integrate Kumo’s Relational Foundation Model with enterprise data systems; contribute to tooling, retrieval, and guardrails
Develop adaptive, multi-step workflows (LLM orchestration, tool use, feedback loops) that continuously refine outputs
Ensure interpretability and evaluation: traceability of steps, confidence scoring, and human-in-the-loop review
Collaborate with PM/design/ML research to turn ambiguous problems into shippable product; instrument, measure, iterate
Demo your work to customers and community, serving as a visible builder and advocate for Kumo RFM
Optimize for latency, cost, and reliability in production environments (serving, caching, tracing, observability)
Qualification
Required
1+ years in ML/AI product development or software engineering (startup or fast-paced product teams)
Hands-on with embeddings, vector databases, and RAG; practical experience evaluating retrieval quality
Strong background in deep learning/transformers/foundation models and LLM orchestration (tool use, planning, memory)
Experience with relational data & SQL; structured reasoning on business datasets
Proficiency in Python and familiarity with data wrangling (Pandas, NumPy)
Strong product sense and collaboration skills—comfortable working with PMs/design and iterating with users
Preferred
Experience as a Founding Engineer or early builder at a startup/innovation pod
Experience with LangChain, LangGraph, LlamaIndex, OpenAI/Anthropic APIs, and multi-agent coordination libraries
Track record building full-stack features (you can dip into frontend/backend/data/infra as needed)
Experience integrating agents with enterprise systems and APIs; designing foundation APIs for tools
Background in knowledge graphs, GNNs, causal inference, or structured reasoning with LLMs
MLOps and cloud (AWS/GCP), model/agent serving, prompt/runtime observability, and eval pipelines
Familiarity with guardrails, safety, and governance for enterprise AI
Company
Kumo
Kumo is an AI model company that provides a platform for anyone to train and run state-of-the-art AI models on their relational data.
51-200 employees
H1B Sponsorship
Kumo has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (8)
2024 (3)
2023 (3)
2022 (2)
Funding
Current Stage
Growth StageTotal Funding
$36.5MKey Investors
Sequoia Capital
2022-09-27Series B· $18M
2022-04-07Series A· $18.5M
Leadership Team
Vanja Josifovski
CEO, Co-founder
Jure Leskovec
Co-Founder and Chief Scientist
Recent News
2025-12-02
Company data provided by crunchbase