Hybrid Staff AI Product Engineer

at Semgrep

• Job title: Staff AI Product Engineer
• Job type: Full Time
• Experience level: Lead
• Salary: $202,000 - $238,000 per year
• Degree requirement: Bachelor's Degree
• Tech skills: Python
• Location requirements: Hybrid, San Francisco, United States

About the role

• AI Product Engineer applying AI/ML tools to build security features at Semgrep. Collaborating with cross-functional teams in a hybrid work environment.

Responsibilities

• Integrate AI platform APIs into the Code product
• Develop and refine LLM prompt chains for real developer use cases
• Experiment with the latest AI/ML advances and determine how they can be productized
• Evaluate and fine-tune ML models using human- and machine-generated data
• Learn directly from users to understand their needs and deliver features that help them secure their code
• Work on major product initiatives end-to-end, from design and prototyping through implementation and deployment
• Contribute to technical discussions, roadmap planning, and mentoring within the team

Requirements

• 8+ years of experience writing production software
• Curiosity and a love of new technologies, especially AI/ML
• Experience experimenting with GPT-4/GPT-5, Codex, Claude, or other LLMs; familiarity with ML algorithms and applied research
• Comfort working in a fast-paced environment where prototypes are rapidly iterated or discarded
• Strong Python skills (experience with other languages a plus)
• Interest in prompt engineering, embeddings, and vector databases
• Excitement about building for customers, iterating fast, and seeing solutions solve real developer problems
• Excellent and proactive communication skills, both verbal and written.

Benefits

• Our compensation package includes equity and benefits in addition to salary.
• We invest in our employees’ well-being and long term success with comprehensive health plans, generous vacation time, 401k, learning stipends, and more.