Senior ML / AI Engineer

Role details

ABOUT THE ROLE Join a small, senior engineering team building the machine learning foundation for consumer software. You will shape data pipelines, models, retrieval systems, and inference infrastructure that power useful, intuitive product experiences, from search and personalization to assistant-like features. WHAT YOU'LL DO - Architect and build machine learning and retrieval systems that support core product experiences. - Design data pipelines to ingest, classify, enrich, and summarize large volumes of user data. - Develop scalable inference infrastructure for real-time and batch workloads. - Select and fine-tune models, build embeddings and retrieval-augmented generation pipelines, and establish evaluation frameworks and quality measures. - Prototype new product ideas, then make successful approaches reliable and production-ready. - Partner with product, design, and engineering to turn complex AI capabilities into polished, easy-to-use features. - Help establish strong machine learning practices across a small, high-ownership team. WHAT WE'RE LOOKING FOR - At least 5 years of software or machine learning engineering experience, with experience in consumer AI or machine learning systems. - Experience building and shipping machine learning product features from early concept to real users, such as personalization, ranking, search, or recommendations. - Hands-on experience developing custom models, including embeddings, recommendation, or personalization models, beyond prompt-based integrations alone. - Strong skills in Python and experience with FastAPI, PostgreSQL, vector databases, embeddings, or related technologies. - Background in applied machine learning, NLP, retrieval, LLMs, or inference systems, with sound judgment about how technical choices affect user experience. - A degree in computer science, mathematics, electrical engineering, or a related field, or equivalent fundamentals. - Progressive experience, comfort working in ambiguity, and experience contributing at a startup or in an early engineering team are valued. COMPENSATION & BENEFITS Base salary is $200,000 to $300,000 USD annually. Visa sponsorship is available. LOCATION On-site in San Francisco, California.