Petra Security Logo

Petra Security

Senior Machine Learning Engineer

Posted Yesterday
Be an Early Applicant
In-Office
San Francisco, CA, USA
Senior level
In-Office
San Francisco, CA, USA
Senior level
Own and evolve real-time machine learning detection systems for account compromise. Design, train, deploy, and monitor models in production at scale; architect systems for rapid growth; work with customers to understand attacks; and help establish engineering standards on a small, high-autonomy team.
The summary above was generated by AI
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 2.5 million users across 10,000+ businesses. 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 $30M+ in revenue in the next 12 months.

 
What You’ll Do

Ownership is everything here. You'll be responsible for our core ML detection platform, and the 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 CTO and Head of Machine Learning, you'll design and build the ML systems that identify account compromise in real time.

  • Take models from research to production. You don't just care about model performance in a notebook. You're energized by measuring how your models behave at scale in a live system with real attackers.

  • Architect for scale. We're growing fast. Nothing excites you more than architecting and implementing an elegant solution that'll handle 10-100x the scale in 6 months.

  • 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
  • 4+ years of machine learning engineering experience, with meaningful time shipping models in production systems at scale.

  • Strong quantitative fundamentals: probability & statistics, linear algebra, anomaly detection, behavioral modeling, NLP.

  • Experience training & deploying models in production: You know how to pick the right model for the job. Just as importantly, you know the limitations of each modeling approach you consider.

  • 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 high-stakes domains like quantitative finance or fraud detection.

  • High agency and high output. You identify the problem, propose the solution, and execute. You're constantly looking for ways to expand your scope and circle of competence.

  • Sharp and scrappy. You learn fast: you ask the right questions and figure it out. You ship fast: you find the 90/10 and don't let perfect be the enemy of done.

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

 
The Team

We're small and high-caliber. Our founders are ex-Abnormal Security, Palo Alto Networks, and Bridgewater. Other early team members include:

  • A Sequoia-backed founder

  • An engineering leader from Anduril

  • Quants from Citadel and Point72

You'll work directly with people who are exceptional at what they do.

 
Benefits and Pay:
  • Competitive salary. We pay top dollar for exceptional engineers.

  • Equity

  • Healthcare, dental, vision

  • Lunch and dinner when working in the office

  • 401K

  • Commuting stipend

  • Equinox discount

  • Relocation support

     

This role is in-person, here in SF.

 

----

Petra is an equal opportunity employer. We do not discriminate on the basis of race, color, religion, sex, sexual orientation, gender identity, national origin, age, disability, veteran status, or any other characteristic protected by applicable law. We are committed to building a diverse and inclusive team and welcome applicants of all backgrounds.

Similar Jobs

5 Days Ago
In-Office or Remote
San Francisco, CA, USA
171K-269K Annually
Senior level
171K-269K Annually
Senior level
Cloud • Information Technology • Productivity • Security • Software • App development • Automation
Develop and productionize machine learning systems for search, retrieval, ranking, recommendations, conversational experiences, and AI agents. Own projects across the ML lifecycle, build evaluation datasets and benchmarks, conduct offline and online evaluations, analyze model quality, and improve system performance. Design scalable architectures addressing reliability, latency, privacy, and cost. Collaborate across product and engineering teams, communicate technical decisions, mentor engineers, and contribute to ML best practices.
Top Skills: Agentic SystemsSparkAWSDatabricksDeep LearningEmbeddingsInformation RetrievalJavaKotlinLarge Language ModelsNlpPythonRagRankingRecommendationsSQLTypescript
10 Days Ago
Easy Apply
Remote or Hybrid
United States
Easy Apply
170K-286K Annually
Senior level
170K-286K Annually
Senior level
Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Build and operate production machine learning systems for Safety AI, including low-latency APIs, data pipelines, model serving, evaluation, monitoring, and rollout infrastructure. Process large-scale camera and telematics data, optimize cloud and edge-to-cloud execution, track model performance and drift, and partner with applied scientists, firmware engineers, platform teams, and product managers to deliver reliable safety features.
Top Skills: SparkAWSAzureC++Ci/CdComputer VisionDockerGCPGoGpusGrafanaInfrastructure As CodeJavaKubernetesMachine LearningMlflowPythonPyTorchRayRay ServeScala
10 Days Ago
Hybrid
Santa Clara, CA, USA
143K-243K Annually
Senior level
143K-243K Annually
Senior level
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Build and operate production machine-learning components for a novel exploitability engine that ranks security work by realistic attack paths. Responsibilities include evidence ingestion, entity resolution, graph construction, probabilistic scoring, testing, evaluation, observability, and data/model reliability. The role involves prototyping AI capabilities, collaborating with product and security teams, contributing to architecture and code reviews, and applying AI safety, security, and guardrails.
Top Skills: Ai AgentsAi EvaluationAi ObservabilityAPIsCloud-Native TechnologiesDatabasesDistributed SystemsEmbeddingsGoGraph SystemsJavaLlmsPythonRagTypescriptVector Search

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account