AI/ML Engineer

About Gem

• Forbes America's Best Startup Employers 2026
• Great Place to Work Certified, 2024
• Fortune Best Workplaces for Millennials, 2023

About the Role

What You'll Do

• 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

What You'll Need

• 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

Nice to Haves

• 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

How We Work

• Local development with Vite boots instantly with hot-reload
• CI finishes in about 10 minutes, and commits deploy to production immediately
• We ship quickly and iterate based on customer feedback

• Reasonable working hours
• Weekly team activities and happy hours
• Regular hackathons where we build experimental features
• Work with a world class team from Meta, Uber, and Dropbox

Benefits

• 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

Role Details

• 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.