Demo

Staff Machine Learning Engineer

Petra Security
San Francisco, CA Full Time
POSTED ON 4/29/2026
AVAILABLE BEFORE 5/27/2026
Who We Are

We’re on a mission to stop the most dominant cyberattacks on the planet.

Cybersecurity titans are born when a new attack vector overwhelms existing defenses. When ransomware ran rampant, CrowdStrike emerged as the category leader for securing user endpoints. Today, AI has made account compromise the #1 cybercrime. LLM-powered phishing messages get past email security, allowing attackers to compromise accounts and exfiltrate sensitive data in minutes. Incumbents can't keep up. They catch these attacks late, or miss them altogether.

Enter Petra, security for when no perimeter remains. We are building an AI-driven Detection and Response platform to catch and remove attackers before they can act.

We've grown very quickly, reaching mid-7-figures in ARR in under a year — we now protect over 1.6 million users across 10,000 businesses. We win 90% of head-to-heads. We price premium, yet have extremely low customer churn. Our customers consistently tell us that Petra is the best product they've ever used. You can see a demo here.

This is a rare chance to join early and help scale to $20M in revenue in the next 12 months.

Our founders are YC W24, ex-Abnormal Security, Palo Alto Networks, and Bridgewater.

What You’ll Do

Ownership is everything here. You'll sit at the core of our detection platform — the ML systems you build are what separate Petra from every other player in this space.

Petra is a premium product and it's everyone's job to make sure it stays that way. That means you won't just train models — you'll talk to customers, understand the attacks they're facing, and think creatively about how to find the next signal that widens the gap between us and second place.

Concretely, you will:

  • Build and evolve our detection models. Working closely with our founding team, you'll help design and build the systems that identify account compromise in real time — spanning behavioral analysis, anomaly detection, and model development .
  • Take models from research to production. You care about the full loop — not just model performance in a notebook, but how it behaves at scale in a live system with real attackers.
  • Architect for scale. We're growing fast. Make calls that let us move quickly today without incurring debt that bites us at 10x the user base.
  • Raise the technical bar. As a senior voice on a lean team, your instincts and opinions shape how we build. You'll establish patterns, push for correctness, and raise the quality of everything around you.

Our stack is Node, TypeScript, React, and Python... for now.

Who You Are

  • 3 years of machine learning engineering experience, with meaningful time shipping models in production systems at scale.
  • Deep expertise in one or more of: anomaly detection, behavioral modeling, NLP, or fraud/abuse detection.
  • Strong engineering fundamentals — you write clean, production-grade code and are comfortable owning a system end to end, not just the modeling layer.
  • Experience working with real-time or streaming data pipelines, feature stores, high-performance databases, and distributed systems. Bonus if you've worked in security, fraud, or identity.
  • Familiarity with LLMs and LLM-based systems in production — you understand their capabilities and limitations and know how to build around them.
  • High agency and high output. You don't wait for a spec — you identify the problem, propose the solution, and execute. You ship fast, find the 90/10, and don't let perfect be the enemy of done.
  • Sharp and scrappy. You learn fast, ask the right questions, and figure it out — whether that's a modeling tradeoff, the right architecture, or the right shortcut.
  • Comfortable in a small-team, high-autonomy environment. The scope of your work is rarely handed to you, and you wouldn't have it any other way.
  • Proud of your craft. You care about building things that work well and hold up over time.

Benefits and Pay:

  • Competitive salary — we pay top dollar for exceptional engineers
  • Equity
  • Fully covered healthcare, dental, vision
  • Lunch, dinner, and snacks at the office
  • Relocation support

This role is in-person, here in SF.

Salary.com Estimation for Staff Machine Learning Engineer in San Francisco, CA
$120,867 to $155,177
If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

What is the career path for a Staff Machine Learning Engineer?

Sign up to receive alerts about other jobs on the Staff Machine Learning Engineer career path by checking the boxes next to the positions that interest you.
Income Estimation: 
$77,900 - $95,589
Income Estimation: 
$101,387 - $124,118
Income Estimation: 
$101,387 - $124,118
Income Estimation: 
$119,030 - $151,900
Employees: Get a Salary Increase
View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

Job openings at Petra Security

  • Petra Security San Francisco, CA
  • Who We Are We’re on a mission to stop the most dominant cyberattacks on the planet. Petra Security runs an AI-powered platform to protect SMBs from account... more
  • 3 Days Ago

  • Petra Security San Francisco, CA
  • Who We Are We’re on a mission to stop the most dominant cyberattacks on the planet. Petra Security runs an AI-powered platform to protect SMBs from account... more
  • 3 Days Ago


Not the job you're looking for? Here are some other Staff Machine Learning Engineer jobs in the San Francisco, CA area that may be a better fit.

  • Databricks San Francisco, CA
  • P-1504 The Applied AI team at Databricks sits at the forefront of advancing GenAI-powered products. Over the past years, we’ve launched Databricks Assistan... more
  • 23 Days Ago

  • Protingent San Francisco, CA
  • Position Title: Staff Machine Learning/AI Engineer Position Description: Protingent Staffing has an exciting direct hire opportunity located in San Francis... more
  • 20 Days Ago

AI Assistant is available now!

Feel free to start your new journey!