Staff ML Engineer

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!

• 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