Staff AI Engineer

Socket.dev; Houston (TX); On-site; USD 177,000 - 210,000; Full time

Job summary

Socket.dev is seeking a Staff AI Engineer to lead the design and delivery of AI/ML systems. This role is ideal for an experienced engineer who thrives on architectural decisions, can confidently own systems end-to-end, and contributes to technical leadership across the team.

This position will be onsite in Houston 50% of the time. You’ll define AI architecture, lead model development and optimization, and tackle integration challenges in a cross-functional setting.

Qualifications

• 7+ years of professional software engineering with 4+ years in AI/ML systems in production.
• Expert Python development for scalable, maintainable systems.
• Deep AWS GenAI knowledge and cloud platform experience.
• Proven track record designing and shipping complex AI systems.
• Experience with cost optimization and model deployment.

Responsibilities

• Lead design and delivery of end-to-end ML/AI systems.
• Define AI architecture and engineering standards.
• Mentor engineers and lead design reviews.
• Collaborate with product and leadership to align AI strategy.

Tools

Docker

Kubernetes

Claude Code

Cursor

Job description

This position will be onsite in Houston 50% of the time.

We’re looking for a Staff AI Engineer to lead the design and delivery of AI/ML systems. This role is ideal for an experienced engineer who thrives on architectural decisions, can confidently own systems end-to-end, and contributes to technical leadership across the team.

In this role, you will work as a key technical leader on a cross-functional team, defining AI architecture, leading model development and optimization, and solving challenging integration problems. You’ll be joining real, in-flight work where reliability, security, and scalability are critical. You’ll establish engineering standards, raise the bar on how we build AI systems, and contribute to the technical decisions that shape how our AI work scales over time.

Why This Role Matters

Every role here contributes directly to that mission. Here, you shape how AI systems integrate into enterprise operations, how teams move at real velocity, and how products create measurable impact for clients and the people they serve. We ship production-ready AI in 30 to 45 days. That pace demands people who take ownership, lead with craft, and care deeply about what they put their name on.

What You’ll Do

Craft & Delivery

• Lead the design and implementation of complex ML/AI systems end-to-end, owning architecture decisions and driving solutions from research through production at scale
• Build and evolve scalable ML platforms, pipelines, and infrastructure that support reliable, repeatable model development and deployment across teams
• Set the standard for AI-forward engineering, using tools like Claude and Cursor with sophistication and helping the team adopt them effectively

Collaboration & Communication

• Partner with product, engineering, and leadership to shape AI strategy and align technical direction with business outcomes
• Translate complex AI tradeoffs, risks, and opportunities into clear narratives that drive decision-making across technical and non-technical stakeholders
• Lead design reviews and technical discussions, raising the bar for engineering rigor and constructive challenge across the team

Leadership & Influence

• Define AI architecture and engineering standards, bringing depth on tradeoffs, long-term implications, and responsible AI practices
• Mentor and grow junior and mid-level engineers, multiplying impact through coaching, code reviews, and pairing on hard problems
• Take ownership of the most ambiguous and highest-stakes pieces of work, driving them through to production with care for reliability, cost, and safety

What You’ll Bring

• 7+ years professional software engineering experience, with 4+ years focused on AI/ML systems in production and deep hands-on experience with generative AI development
• Expert software engineering background (Python or similar) with strong design sensibilities for scalable, maintainable systems
• Deep expertise with cloud platforms, including in-depth understanding of AWS services and AWS GenAI offerings
• Proven track record designing and shipping complex agentic systems in production environments
• Mastery of AI frameworks and orchestration tools
• Strong experience with evaluation frameworks and observability tools for LLM apps, including building these capabilities where they don’t yet exist
• Deep understanding of AI safety, responsible AI principles, prompt injection defenses, and PII handling
• Extensive experience building RAG pipelines: chunking strategies, embedding models, vector databases, and advanced retrieval techniques
• API design experience, including architecting and integrating with internal and third‑party services at scale
• Advanced cost optimization expertise: token economics, caching strategies, model routing, quantization
• Strong working knowledge of Docker and Kubernetes for containerized deployments
• Demonstrable, day-to-day usage and expert knowledge of AI-forward coding tools such as Claude Code and Cursor

The below range reflects the range of possible compensation for this role at the time of this posting. We may ultimately pay more or less than the posted range. This range may be modified in the future. An employee’s position within the salary range will be based on several factors including, but not limited to, relevant education, qualifications, certifications, experience, skills, seniority, geographic location, performance, shift, travel requirements, sales or revenue-based metrics, any collective bargaining agreements, and business or organizational needs. The salary range for this role is $-177,375 - $209,625 USD.

We offer a comprehensive package of benefits including paid time off, medical/dental/vision insurance and 401(k) to eligible employees.

An offer of employment may be conditional upon successful completion of a background check in accordance with local legislation and our candidate privacy notice. Your current employer will not be contacted without your permission. We are committed to ensuring equal employment opportunities for all job applicants and employees. Employment decisions are based upon job-related reasons regardless of an applicant’s race, color, religion, sex, sexual orientation, gender identity, age, national origin, disability, marital status, genetic information, protected veteran status, or any other status protected by law.