Staff ML Engineer
Prophecy
| Company | Prophecy |
| Category | Engineering |
| Location | Palo Alto, CA |
| Remote | On-site (inferred) |
| Employment | Not stated |
| Level | Not stated |
| Salary | USD 250k–350k |
| Posted | 6 Aug 2026 |
| Last verified | 14 Aug 2026 |
| Source | The employer's own careers page (company_site) |
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.
Must-Have Skills
Knowledge Graph Expertise
• Expertise in building and managing Knowledge Graphs, including entity extraction, linkage, and ontology design
AI Systems Integration
• Experience integrating Knowledge Graphs with AI systems for retrieval, reasoning, and agent context (e.g., enriching with LLM)
Technical Leadership
• Tech lead experience: Owning end-to-end KG projects, from design to deployment, with leadership in cross-functional teams
Full-Stack Development Skills
• Versatile software developer: fluency in Python, REST APIs, microservices in public cloud; some experience with Go, AWS, Kubernetes, Java, Scala
Graph Database Mastery
• Hands-on experience in developing, optimizing, and operating Neo4j or similar graph databases
Builder Mentality
• 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
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
• FSA/HSA, dental, vision, and more
High-Impact Work
• Ability to have your fingerprint on an innovative platform used by world-class enterprises
• 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!
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
Tech stack
• Python
• Machine Learning
• PyTorch
• neo4j
• SQL
• Scala