Staff ML Engineer

Prophecy; Palo Alto (CA); Hybrid; USD 250,000 - 350,000; Full time

Benefits offered by this job

Health coverage

Wellness stipend

Flexible PTO

Professional development

Long-term disability

Dental & vision

Job summary

Prophecy is seeking a Staff Machine Learning Engineer specializing in Knowledge Graphs in San Francisco, hybrid. The role leads end-to-end KG projects, builds scalable graph systems, and integrates AI agents with the data platform.

You will shape enterprise data prep and insights with cutting-edge tech. The position offers a base salary, equity, and comprehensive benefits, and requires strong Python, AI, cloud, and graph database skills.

Qualifications

• Knowledge Graphs: entity extraction, linkage, ontology design.
• KG integration with AI systems for retrieval and context.
• Tech lead experience delivering end-to-end KG projects.
• Proficient in Python and REST APIs; cloud-native microservices.
• Experience with Go, AWS, Kubernetes, Java, Scala.
• Hands-on with graph databases like Neo4j.

Responsibilities

• Own end-to-end KG projects from design to deployment.
• Lead cross-functional teams and drive delivery.
• Design and scale knowledge graph systems powering data prep and analysis.
• Collaborate with AI agents to integrate enterprise data infrastructure.

Skills

Knowledge graphs

AI integration

Technical leadership

Python

REST APIs

Cloud microservices

Go

AWS

Kubernetes

Java

Scala

Tools

Neo4j

PyTorch

Python

SQL

Neo4j/Graph DBs

Job description

STAFF MACHINE LEARNING ENGINEER Knowledge Graph San Francisco, CA (Hybrid) Senior to Staff Level Build the AI-Native Data Revolution with Knowledge Graphs

About Prophecy: Prophecy is the leader in AI-native data preparation and analysis, revolutionizing how the world's top enterprises turn data chaos into reliable insights. We introduce the AI-native data lifecycle (generate, refine, deploy) where our industry-leading AI agents and humans work hand-in-hand in visual and document interfaces to analyze, transform, and prepare data, shipping trusted insights at enterprise scale. Don't miss the rocket ship—join Prophecy and build the next data revolution.

Compensation Package: $250,000 - $350,000 Base Salary, plus equity, benefits, and comprehensive perks

About the Role: As a Staff Machine Learning Engineer specializing in Knowledge Graphs, you'll be at the forefront of Prophecy's AI-native data platform. You'll design, build, and scale knowledge graph systems that power intelligent data preparation and analysis, integrating cutting‑edge AI agents with enterprise‑scale data infrastructure. This is a high‑impact technical leadership role where you'll own end‑to‑end projects and shape the future of how enterprises understand and leverage their data.

Requirements

• Expertise in building and managing Knowledge Graphs, including entity extraction, linkage, and ontology design
• Experience integrating Knowledge Graphs with AI systems for retrieval, reasoning, and agent context (e.g., enriching with LLM)
• Tech lead experience: Owning end-to-end KG projects, from design to deployment, with leadership in cross‑functional teams
• Versatile software developer: fluency in Python, REST APIs, microservices in public cloud; some experience with Go, AWS, Kubernetes, Java, Scala
• Hands‑on experience in developing, optimizing, and operating Neo4j or similar graph databases
• Demonstrated experience taking ideas to production with a strong bias toward action and delivery

What Will Make You Stand Out

• Advanced ML/LLM Work — ML/LLM work in code generation (e.g., Codex, text‑to‑SQL), semantic extraction, or knowledge graphs (e.g., Neo4j, Neptune)
• Big Data Experience — Experience with big‑data engines like Spark for processing massive datasets
• Compiler Development — Compiler development for languages like SQL, Python, or Scala
• Production ML Optimization — Optimization of ML models for low‑latency, high‑throughput production use
• Open Source Contributions — Contributions to open‑source AI/ML projects (e.g., Hugging Face, PyTorch)
• Retrieval Systems Expertise — Expertise in retrieval systems or vector databases (e.g., Pinecone, Weaviate)
• Strategic Technical Decision‑Making — Skill in evaluating technology options and driving build/buy decisions

Tech stack

• Python
• Machine Learning
• PyTorch
• neo4j
• SQL
• Scala

Position Details

• Seniority Level: Senior to Staff Level
• Experience Required: 5+ years in industry
• Base Compensation: $250,000 - $350,000
• Location: San Francisco, CA
• Work Model: Hybrid

Benefits and Perks

• Comprehensive Health Coverage Prophecy covers 99% of employee health insurance premiums and 75% for dependents
• Remote Work & Wellness Stipend $200 per month towards remote work, wellness, gyms, massages, facials, and more!
• Flexible Time Off Flexible PTO to maintain work‑life balance
• Professional Development Professional development allowance to support your continuous growth
• Additional Benefits Company-sponsored Long Term Disability and Life Insurance
• Additional Benefits FSA/HSA, dental, vision, and more
• High-Impact Work Ability to have your fingerprint on an innovative platform used by world‑class enterprises
• High-Impact Work End‑to‑end ownership of your projects with technical leadership opportunities
• *Benefits and perks may vary per country

Why Join Prophecy?

• Revolutionary AI‑Native Platform Work on cutting‑edge AI agents and knowledge graph systems, transforming enterprise data preparation
• Technical Leadership Role Own end‑to‑end knowledge graph projects and lead cross‑functional teams at Staff level
• Enterprise Scale Impact Your work will power data insights for the world’s top enterprises
• Cutting‑Edge Tech Stack Work with LLMs, knowledge graphs, graph databases, and the latest AI/ML technologies
• Competitive Compensation $250K-$350K base salary plus equity, comprehensive benefits, and generous perks

Ready to revolutionize enterprise data with Knowledge Graphs? Join Prophecy's rocket ship and build the next data revolution.