AI/ML Engineer

About Gem

About the Role

Build multi-step reasoning systems that understand nuanced recruiter queries like 'senior ML engineers who've worked at YC companies and know PyTorch' – currently processing 800M+ profiles
Develop intelligent search that understands recruiter intent and surfaces the right candidates instantly
Ship rapidly and iterate by moving from prototype to production, measuring performance, and refining based on live product data and user feedback
Collaborate closely with product managers, designers, and engineers in our SF office to create cohesive AI-driven experiences
Share ML expertise with teammates and help establish best practices for model development and evaluation

Experience applying AI/ML in production environments, especially with LLMs
Hands-on knowledge of retrieval and search systems, including query understanding, ranking, and relevance optimization
Familiarity with vector databases and embedding models for AI applications
Understanding of retrieval architectures and their implementation
Pragmatic approach to AI. You find the right balance between cutting-edge techniques and scalable, maintainable solutions

Problem-solving: You approach problems holistically, starting with clear understanding of context. You navigate ambiguity well, decompose complex problems into clean solutions, and think critically about real-world impact
Shipping quality products: You partner closely with product and design to craft experiences that feel effortless. You understand the balance between craft, speed, and business impact
Communication and collaboration: You explain technical concepts clearly to both engineers and non-technical stakeholders. You engage thoughtfully in disagreements and compromise when needed
Velocity and ownership: You're excited by quick experimentation and continuous improvement. You take pride in shipping features from start to finish
User focus: You care about business impact and prioritize accordingly. You think critically about how your work shapes real people's daily workflows

Prior startup experience (especially 0→1 product)
Prompt and context engineering skills or experience with LLM fine-tuning
Background in information retrieval, NLP, or recommendation systems
Experience implementing feedback loops to improve search quality over time
Track record of shipping AI features that balance product craftsmanship with technical rigor

10-year stock option exercise window
Flexible Time Off and 16 paid holidays (including company wellness days)
Best-in-class medical, dental, and vision coverage
$1,200 annual learning and development stipend
16 weeks paid parental leave for all parents, plus $1,500 new-parent perk and flexible return-to-work options

Location: This role is based 3 days per week out of our San Francisco HQ and is not eligible for full-time remote work.
Compensation: The annual cash compensation range for this position is $190,000–$260,000 based on level in addition to equity & benefits.