Staff AI Engineer

Posted Yesterday
Hiring Remotely in USA
Remote
275K-300K Annually
Expert/Leader
Security • Software
Cobalt provides offensive security testing for security and development teams.
The Role
Lead the technical direction, development, evaluation, security, observability, and cost optimization of production LLM agent systems. Build shared agent infrastructure, improve reliability and fault tolerance, establish testing and rollout standards, and partner with engineering, product, research, and security teams. Mentor engineers, support AI coding-agent adoption, investigate incidents, and guide model selection and production readiness across teams.
Summary Generated by Built In
Description

Cobalt runs AI agents in production that test real customer applications. We’re looking for a Staff AI Engineer who has built and operated production agent systems and can help us make them more reliable, secure and cost-efficient.

You’ll set the technical direction for how we design, evaluate, ship and operate agents, and stay hands-on building the foundations our teams use. You’ll work across engineering, product, research and security, mentoring engineers and helping teams turn promising prototypes into dependable production systems.

Your primary focus will be our agent platform and production practices. You’ll also help engineering teams improve how they build software with AI coding agents.

What You’ll Do
  • Set technical direction for agent systems across teams, including orchestration, tool use, context and memory management, human oversight and failure recovery.
  • Design, build and operate shared foundations: evaluation frameworks, agent observability and tracing, execution controls, and model access with routing and failover where needed.
  • Establish evaluation practices that measure task completion, accuracy, false positives, security and cost per successful task. Work with domain experts to build representative test cases and turn production failures into regression tests.
  • Make agent workflows resilient to tool failures, timeouts, rate limits and partial execution, with clear limits on retries, runtime and spend.
  • Work with security and platform teams to build in scoped permissions, customer data isolation, audit trails and safe execution of tools and code.
  • Establish practical engineering standards: evals in CI, prompt and model versioning, regression testing for non-deterministic systems, cost and latency budgets, and staged rollouts with rollback paths.
  • Lead model selection and cost optimization. Benchmark models against real workloads and make informed trade-offs between quality, latency and cost.
  • Partner with teams on production readiness, incident investigation and improvements to agent reliability.
  • Help teams use AI coding agents effectively across planning, implementation, review and testing. Measure the impact on delivery speed and software quality.
  • Mentor engineers and technical leads through design reviews, pairing, internal talks and written guidance. Help teams adopt shared capabilities and improve them based on feedback.
  • Work with product, research and security leaders to turn business priorities into a technical roadmap with clear outcomes.
You Must Have
  • Significant software engineering experience, typically 8–10+ years, with a track record of technical leadership across teams.
  • Hands-on experience building and operating LLM-based agents in production, with a clear understanding of their scale, constraints and business outcomes.
  • A track record of improving agent reliability: building evaluations, investigating non-deterministic failures and improving success rates against measurable goals.
  • Strong knowledge of agent architectures, including tool use, orchestration, planning, context and memory management, and the judgment to know when simpler workflows are sufficient.
  • Experience running LLM workloads through managed cloud model platforms or direct provider APIs, including rate limits, failure handling and cost tracking.
  • Strong backend and distributed systems fundamentals, including APIs, asynchronous workflows, concurrency and fault tolerance.
  • Experience with production observability, incident investigation and safe rollout of system changes.
  • An understanding of security risks in agent systems, including prompt injection, unsafe tool execution and exposure of sensitive data.
  • Evidence of impact beyond your own code: shared platforms other teams adopted, standards that improved outcomes or engineers you helped grow.
Tools and Practices You’ve Likely Used

We care more about your depth of experience and engineering judgment than experience with any specific tool.

  • Agent orchestration: Framework-based or custom orchestration, tool calling and structured outputs.
  • Tool integration: Model Context Protocol (MCP), APIs and permission-controlled tool execution.
  • Evaluations: Automated evaluation harnesses, domain-specific test datasets and regression testing in CI.
  • Observability: Distributed tracing, structured logs and metrics for multi-step agent workflows.
  • Model access: Managed cloud model platforms, direct provider APIs and multi-provider gateways.
Nice To Have
  • Experience with GCP, Go or Kubernetes.
  • Experience building sandboxed environments for agents that execute untrusted code or interact with external systems.
  • Offensive security, application security or security tooling experience.
  • Experience introducing AI coding tools across engineering teams and measuring their impact.
  • Public writing, talks or open-source contributions related to agent engineering.
Why You Should Join Us
  • Grow in a passionate, rapidly expanding industry operating at the forefront of the Pentesting industry 
  • Work directly with experienced senior leaders with ongoing mentorship opportunities
  • Earn competitive compensation and an attractive equity plan
  • Save for the future with a 401(k) program (US) or pension (EU) 
  • Benefit from medical, dental, vision and life insurance (US) or statutory healthcare (EU)
  • Leverage stipends for:
  • Wellness
  • Work-from-home equipment & wifi
  • Learning & development
  • Make the most of our flexible, generous paid time off and paid parental leave 
Pay Range Disclosure 

Cobalt is committed to fair and equitable compensation practices. The salary range for this role is ($275,000 - $300,000) + equity + benefits. A candidate’s salary is determined by various factors including, but not limited to, relevant work experience, skills, and certifications.  The salary range may differ in other states and may be impacted by proximity to major metropolitan cities. 

Cobalt (the "Company") is an equal opportunity employer, and we want the best available persons for every job. The Company makes employment decisions only based on merit. It is the Company's policy to prohibit discrimination in any employment opportunity (including but not limited to recruitment, employment, promotion, salary increases, benefits, termination and all other terms and conditions of employment) based on race, color, sex, sexual orientation, gender, gender identity, gender expression, genetic information, pregnancy, religious creed, national origin, ancestry, age, physical/mental disability, medical condition, marital/domestic partner status, military and veteran status, height, weight or any other such characteristic protected by federal, state or local law. The Company is committed to complying with all applicable laws and providing equal employment opportunities. This commitment applies to all persons involved in the operations of the Company regardless of where the employee is located and prohibits unlawful discrimination by any employee of the Company.

Cobalt is an E-Verify employer. E-Verify is an Internet-based system operated by the Department of Homeland Security (DHS) in partnership with the Social Security Administration (SSA). It allows participating employers to electronically verify the employment eligibility of their newly hired employees in the United States.

Skills Required

  • Significant software engineering experience, typically 8–10+ years, with technical leadership across teams.
  • Hands-on experience building and operating LLM-based agents in production.
  • Experience improving agent reliability through evaluations, non-deterministic failure investigation, and measurable success-rate improvements.
  • Strong knowledge of agent architectures, including tool use, orchestration, planning, context management, and memory management.
  • Experience running LLM workloads through managed cloud model platforms or direct provider APIs, including rate limits, failure handling, and cost tracking.
  • Strong backend and distributed systems fundamentals, including APIs, asynchronous workflows, concurrency, and fault tolerance.
  • Experience with production observability, incident investigation, and safe system rollouts.
  • Understanding of agent security risks, including prompt injection, unsafe tool execution, and sensitive-data exposure.
  • Evidence of impact beyond individual code contributions, such as shared platforms, engineering standards, or mentoring.
  • Experience with GCP, Go, or Kubernetes.
  • Experience building sandboxed environments for agents executing untrusted code or interacting with external systems.
  • Offensive security, application security, or security tooling experience.
  • Experience introducing AI coding tools across engineering teams and measuring their impact.
  • Public writing, talks, or open-source contributions related to agent engineering.

Cobalt Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Cobalt and has not been reviewed or approved by Cobalt.

  • Strong & Reliable Incentives — Pay is considered competitive in sales roles, with on-target structures and accelerators that can yield substantial upside for high performers. Feedback suggests incentive design rewards attainment and offers meaningful earning potential.
  • Leave & Time Off Breadth — Time off provisions include flexible or unlimited PTO, paid sick days, paid volunteer time, bereavement leave, and a day off for birthdays. Feedback suggests generous parental leave complements these policies.
  • Equity Value & Accessibility — Equity is positioned as a core part of total compensation, with stock options and an employee stock purchase plan noted. Employees are offered an attractive equity plan alongside base pay.

Cobalt Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Arlington, VA
150 Employees
Year Founded: 2013

What We Do

Cobalt’s Pentest as a Service (PtaaS) platform is modernizing traditional pentesting. By combining a SaaS platform with an exclusive community of testers, we deliver the real-time insights you need to remediate risk quickly and innovate securely.

Why Work With Us

We stand at the forefront of cybersecurity innovation. As the leader in Pentesting as a Service, we empower companies to innovate securely and operate with confidence. Our fully remote team is united by a shared mission: to be the undisputed pentest leader. We’re a purpose-driven company where your work has real impact and your career can thrive.

Gallery

Gallery

Similar Jobs

Zscaler Logo Zscaler

Artificial Intelligence Engineer

Cloud • Information Technology • Security • Software • Cybersecurity
Easy Apply
Remote or Hybrid
Santa Clara, CA, USA
8697 Employees
180K-225K Annually

General Motors Logo General Motors

Artificial Intelligence Engineer

Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
Remote or Hybrid
2 Locations
165000 Employees
Remote
United States
575 Employees
150K-190K Annually

Deepgram Logo Deepgram

Sales Engineer

Artificial Intelligence • Machine Learning • Natural Language Processing • Software • Conversational AI
Remote
USA
150 Employees
185K-230K Annually

Similar Companies Hiring

Ford Energy Thumbnail
Automotive • Software • Energy • Utilities • Manufacturing • Renewable Energy
Glendale, Kentucky
55 Employees
Revel Thumbnail
Aerospace • Hardware • Robotics • Software
Marina Del Rey, California
70 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
33 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account