AI Staff Software Engineer
Staff AI Software Engineer builds and scales AI systems including PromptQL AI assistant, data engine, and runtime infrastructure for enterprise reliability. Requires 6+ years experience with AI/ML, distributed systems, LLMs, Python fluency, and technical leadership.
About the job
Role Requirements
- 6+ years of experience as a software engineer, with at least 1+ years building AI or ML-powered systems
- Designed and shipped complex distributed systems at scale, and know how to optimize for performance, cost, and reliability
- Worked directly with large language models (LLMs) in production
- Fluent in Python and/or a systems language, and can move comfortably between infrastructure, APIs, and core product logic
- Comfortable leading technically: writing high-leverage design docs, setting engineering direction, and mentoring other senior engineers
- Biased toward action and can operate with high autonomy; know how to unblock yourself and others
- Care about quality, safety, and performance; bring a systems-level perspective to how engineering decisions impact the product and user
Responsibilities
- Write production code, design architecture, optimize performance, shape infrastructure, and contribute to safety mechanisms
- Build core stack, rapidly design and implement features across the PromptQL AI assistant, the PromptQL data engine, and the PromptQL runtime infrastructure
Compensation
- Salary $250,000 - $500,000
- Annual bonus
- Large equity package
Skills
Python, LLMs, Distributed Systems, Machine Learning, Cloud Infrastructure, APIs, Systems Programming
Similar jobs
ML Engineering jobs
Haus
Seattle, WAStaff Machine Learning Engineer building and operating production ML systems for causal marketing measurement, optimization, and planning. The role requires deep statistical and machine learning expertise, production programming experience, cross-functional collaboration, and technical mentorship.
Scale AI
San Francisco, CAStaff engineer responsible for designing and scaling the infrastructure, execution environments, verifiers, and tooling used to train and evaluate AI agents. Requires 8+ years of software engineering experience, strong Python and distributed-systems expertise, and familiarity with sandboxing, high-throughput systems, and LLM workflows.
Airbnb
United States
Senior Staff ML Engineer fine-tunes and optimizes state-of-the-art LLMs for Airbnb's customer support AI products, including AI assistants and autonomous agents. Partners cross-functionally to productionize models at scale. Requires PhD and 10+ years experience with PyTorch.
United States
Leads the technical direction of large-scale ML infrastructure for embedding, recommendation, and personalization systems. The role requires 8+ years of ML engineering experience, expertise in deep learning and distributed training, and strong leadership across research, infrastructure, and production deployment.
Snowflake
Bellevue, WA
Leads the roadmap and technical vision for Snowflake Feature Store, building reliable, high-performance machine learning platform capabilities and supporting technical execution across partner teams. Requires 10+ years of experience with data-serving infrastructure or ML platforms, plus Java and Python expertise.