Senior ML / AI Engineer

About the Role

Join a small, senior engineering team building the machine learning foundation for consumer software products. You will shape data pipelines, models, retrieval systems, and inference infrastructure that make AI-powered experiences useful and intuitive for everyday users.

What You'll Do

• Architect and build machine learning and retrieval systems for core product experiences.
• Design data pipelines that ingest, classify, enrich, and summarize large volumes of user data.
• Build scalable infrastructure for real-time and batch inference.
• Guide model selection, fine-tuning, embeddings, retrieval-augmented generation, and evaluation frameworks.
• Prototype product ideas and turn successful experiments into reliable systems.
• Partner with product, design, and engineering to deliver polished user-facing features.
• Establish strong practices for system quality, measurement, and failure analysis.

What We're Looking For

• At least 5 years of software or machine learning engineering experience, with a background in consumer AI or machine learning systems.
• Experience shipping machine learning features to real users, such as personalization, ranking, search, or recommendations.
• Experience developing custom models, including embeddings, recommendation, or personalization models, beyond prompt-based integrations alone.
• Strong experience with applied machine learning, LLMs, retrieval, classification, NLP, or related product systems.
• Familiarity with Python, FastAPI, PostgreSQL, vector databases, and embeddings.
• Experience with data pipelines, training and evaluation workflows, inference infrastructure, and end-to-end product development.
• A degree in computer science, mathematics, electrical engineering, or a related field, or comparable technical background.
• Comfort working in an early-stage or entrepreneurial environment and collaborating across product and design.

Compensation & Benefits

Base salary

$200,000 to $300,000 annually. Visa sponsorship is available.

Location

On-site in San Francisco, California, United States.