Skip to content
DatabricksDatabricksIllinois

AI Engineer

Build and ship AI-powered analytics agents, LLM-driven workflows, and intelligence features for Databricks' internal GTM platform (Customer Zero). Requires 7+ years software/AI engineering experience, strong Python/SQL, hands-on LLM/RAG/agents experience, and fluency with AI coding tools like Claude.

117k – 202k/yr
Remote7+ YOEML Engineering

About the role

What you'll do

  • Build and ship features for AI-powered surfaces: Genie analytics agents, LLM-assisted feedback and enablement workflows, and in-app intelligence inside the GTM Hub.
  • Develop and integrate agentic workflows — prompts, tools/MCP integrations, retrieval, and evaluation — that turn business questions into reliable, grounded answers.
  • Write Python and SQL against our Databricks lakehouse (Unity Catalog, Delta) and Lakebase (Postgres) services that back the apps.
  • Use AI coding tools (Claude Code, agentic skills) as a core part of your workflow, and help build the reusable skills and harnesses that make the whole team faster.
  • Contribute to evaluation and quality: help measure and improve the accuracy of our AI outputs (eval sets, scorecards, regression checks).
  • Partner with data engineers, app engineers, and strategy/ops stakeholders to take work from idea → ticket → PR → production.
  • Write clean, well-documented, tested code and participate in code review (human and AI-assisted).

What we look for

  • Bachelor's degree in Computer Science, Data Science, or a related field — or equivalent practical experience.
  • 7+ years of software and AI engineering related experience.
  • Solid fundamentals in Python and SQL.
  • Hands-on exposure to LLMs / generative AI — prior work building with APIs, prompts, RAG, Knowledge graphs & agents.
  • Genuine fluency with AI developer tools (e.g., Claude, Cursor) and a desire to push how far AI-assisted engineering can go.
  • Strong problem-solving, curiosity, and a bias to ship and iterate; comfortable with ambiguity in a fast-moving team.
  • Clear written and verbal communication; works well with both engineers and non-technical stakeholders.

Nice to have

  • Experience with the Databricks platform (AI Gateways, notebooks, jobs, Unity Catalog, Databricks Apps, Genie) or another cloud data platform.
  • Building or evaluating agentic, RAG systems; familiarity with MCP, tool-calling, or eval frameworks.
  • Full-stack exposure (React/TypeScript front end, FastAPI/Python back end) or data engineering (dbt, Spark, medallion architectures).
  • Experience with Postgres/OLTP, CI/CD (Databricks Asset Bundles, GitHub Actions), or analytics/BI.

Skills

PythonSQLLLMsGenerative AIRAGagentsDatabricksunity catalogDelta LakePostgresReactTypeScriptFastAPISparkdbt

Similar roles

ML Engineering jobs
Bloomerang

Sr. Software Engineer

BloomerangUnited States

Sr. Software Engineer building and owning Bloomerang's agentic AI harness, including workflows, orchestration, guardrails, evals, and model integrations. Near-term focus on backend platform services (auth/identity in C#/.NET). Requires 8+ years experience, AI fluency, and ownership of reliable AI-driven development for nonprofit impact.

115k – 191k/yrRemote8+ YOEML Engineering
PointClickCare

Sr Application Engineer

PointClickCareUnited States

Senior Application Engineer building and deploying AI Agents with Salesforce Agentforce, intelligent automations, and custom Salesforce applications. Requires 5+ years experience, specific Salesforce certifications (Platform Developer I/II, Advanced Admin, Agentforce Specialist), strong coding and DevOps skills, and mentoring ability.

120k – 135k/yrRemote5+ YOEML Engineering
LiveKit

Senior Software Engineer, Agents

LiveKitUnited States

Designs and implements core Python framework components for building real-time AI agents that see, hear, and speak. Owns features end-to-end with strong API design skills and experience in production Python systems.

120k – 250k/yrRemoteML Engineering
Applied Intuition

Senior Machine Learning Data Engineer

Applied IntuitionSunnyvale, CA

Build and optimize edge and cloud data pipelines for ML perception models, including MLOps tooling, foundation model integration for automated labeling, and customer interaction with DoD stakeholders. Requires 5+ years experience with ML infrastructure, data-centric AI, and large-scale GPU jobs.

125k – 220k/yrOn-site5+ YOEML Engineering
Kraken

Senior AI Engineer

KrakenUnited States

Build and ship AI agent orchestration workflows, integrations, and shared infrastructure that power automated growth across paid media, lifecycle, organic, CRO, incentives and more at Kraken. Requires 3+ years building agentic systems or automation with LLM APIs, strong API integration skills, and growth/marketing context.

127k – 254k/yrRemote3+ YOEML Engineering