Cobalt Logo

Cobalt

Staff AI Engineer

Posted 5 Days Ago
Remote
Hiring Remotely in USA
275K-300K Annually
Expert/Leader
Remote
Hiring Remotely in USA
275K-300K Annually
Expert/Leader
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.
The summary above was generated by AI
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.

Similar Jobs

Yesterday
Remote or Hybrid
CA, USA
187K-252K Annually
Senior level
187K-252K Annually
Senior level
Artificial Intelligence • Cloud • Information Technology • Sales • Security • Software • Cybersecurity
Designs, builds, and ships production LLM and agentic AI systems for cybersecurity products. Leads architecture, model strategy, fine-tuning, post-training, inference optimization, evaluation, and technical roadmaps. Establishes engineering practices, mentors engineers, leads design reviews, prototypes critical model and training code, and partners cross-functionally to deliver scalable, cost-effective AI solutions.
Top Skills: Agentic AiClassical Machine LearningDeepspeedDistributed SystemsDpoFsdpGrpoLlmsLoraPythonPyTorchQloraQuantizationRetrieval-Augmented GenerationSglangTensorrt-LlmTransformer ArchitecturesVllm
5 Days Ago
Easy Apply
Remote or Hybrid
Easy Apply
180K-225K Annually
Senior level
180K-225K Annually
Senior level
Cloud • Information Technology • Security • Software • Cybersecurity
Build and deploy secure, production-grade AI solutions for go-to-market teams using retrieval, structured-data querying, and agentic workflows. Partner with GTM leaders to define AI strategy, roadmaps, and build-versus-buy decisions. Establish monitoring, evaluation, security, and ROI frameworks tied to revenue outcomes. Drive adoption through enablement and feedback loops, develop reusable AI architecture and data connectors, improve engineering standards, and mentor other engineers.
Top Skills: A/B TestingAgent-To-Agent (A2A) FrameworksAgentic WorkflowsCloud PlatformsGraphragKnowledge GraphsLlm-As-JudgeModel Context Protocol (Mcp)PythonRegression TestingRetrieval-Augmented Generation (Rag)
8 Days Ago
Remote
United States
150K-190K Annually
Expert/Leader
150K-190K Annually
Expert/Leader
Software
Designs and deploys AI-powered features and scalable AI/ML services across SaaS products. Responsibilities include translating business problems into AI solutions, integrating large language models, preparing data, evaluating model performance, writing Python, building secure deployment infrastructure, and collaborating with product, data, and cloud operations teams. The role requires production AI/ML experience, scalable data infrastructure expertise, AWS familiarity, and modern DevOps practices.
Top Skills: Ai/Ml Evaluation ToolingAWSDevOpsLarge Language Models (Llms)Llm FrameworksPythonSaaS

What you need to know about the Los Angeles Tech Scene

Los Angeles is a global leader in entertainment, so it’s no surprise that many of the biggest players in streaming, digital media and game development call the city home. But the city boasts plenty of non-entertainment innovation as well, with tech companies spanning verticals like AI, fintech, e-commerce and biotech. With major universities like Caltech, UCLA, USC and the nearby UC Irvine, the city has a steady supply of top-flight tech and engineering talent — not counting the graduates flocking to Los Angeles from across the world to enjoy its beaches, culture and year-round temperate climate.

Key Facts About Los Angeles Tech

  • Number of Tech Workers: 375,800; 5.5% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Snap, Netflix, SpaceX, Disney, Google
  • Key Industries: Artificial intelligence, adtech, media, software, game development
  • Funding Landscape: $11.6 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Strong Ventures, Fifth Wall, Upfront Ventures, Mucker Capital, Kittyhawk Ventures
  • Research Centers and Universities: California Institute of Technology, UCLA, University of Southern California, UC Irvine, Pepperdine, California Institute for Immunology and Immunotherapy, Center for Quantum Science and Engineering

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account