Staff Engineer, AI Simulation & Infrastructure

Engg; San Francisco (CA); On-site; USD 190,000 - 280,000; Full time

Job summary

Patronus AI is seeking a hands-on Staff Engineer to build scalable simulation infrastructure and ML systems. You’ll own end-to-end delivery across frontend, backend, and ML workloads, collaborating with researchers to translate ambitious AI ideas into robust production platforms.

You should ship code with strong product judgment, operate effectively in ambiguity, and mentor teammates while advancing tooling for agent environments and evaluation.

Qualifications

• Track record of shipping non-trivial software end-to-end as an individual contributor.
• Strong fundamentals with depth in backend/infrastructure, frontend/product engineering, or ML systems.
• Experience building production systems in Python, Go, or TypeScript; quick to adapt to new stacks.
• Experience with modern LLMs and agents at the application level and tool calling concepts.
• Good architectural judgment around design, reliability, and operational complexity.

Responsibilities

• Build agent environments and simulations end-to-end, including frontend interfaces and APIs.
• Develop infrastructure for agent gym, orchestration, sandboxing, and benchmarking.
• Create internal platforms and developer tools for researchers and engineers.
• Operate ML infrastructure, including model deployment and GPU workloads.
• Own systems from idea to production and iterate based on performance.
• Collaborate with researchers to productionize experiments and scale tooling.

Skills

Shipping end-to-end software

Backend/infrastructure depth

Frontend/product engineering

ML systems

Python

Go

TypeScript

LLMs and agents

System design judgment

Independent work in ambiguity

AI tooling fluency

Education

BS/MS/PhD in Computer Science, Machine Learning, Software Engineering, or related field

Tools

React

TypeScript

Next.js

Python

Relational databases

APIs / API frameworks

ML tooling

Job description

Patronus AI is seeking a hands-on Staff Engineer to build scalable simulation infrastructure and ML systems. You’ll own end-to-end delivery across frontend, backend, and ML workloads, collaborating with researchers to translate ambitious AI ideas into robust production platforms.

You should ship code with strong product judgment, operate effectively in ambiguity, and mentor teammates while advancing tooling for agent environments and evaluation.