Staff Engineer, Agentic AI

About the Role

Lead the development of agent intelligence that helps mechanical engineers complete complex, multi-step workflows across desktop engineering software. As a hands-on technical lead on a small AI engineering team, you will shape the agent architecture, connect user research with product development, and improve the reliability and cost efficiency of real-world workflows.

What You'll Do

• Own agent task success metrics, establish performance baselines, and systematically improve completion rates.
• Build rigorous, reproducible evaluation infrastructure grounded in validated user stories and real engineering workflows.
• Set per-task token budgets and track the cost of completed workflows.
• Work with researchers and engineering domain experts to map, interview about, and validate user workflows.
• Translate validated user stories into testable evaluations and prioritize workflow coverage by customer value and technical feasibility.
• Make architecture decisions across tool calling, state management, error recovery, model routing, and context management.
• Set technical direction for a small team, review designs and code, unblock teammates, and contribute production code.
• Collaborate with product, integrations, and customers to align agent behavior with real-world use.

What We're Looking For

• At least 7 years of software engineering experience, including 2 or more years building and shipping LLM-based agents that take real-world actions.
• Deep experience with LLM application architecture, including model selection, context management, retrieval, tool calling, and orchestration.
• Strong Python skills and familiarity with function calling, tool APIs, tracing or observability, and evaluation tooling.
• Experience building benchmarks for task completion, cost efficiency, and failure analysis.
• Hands-on technical leadership and code review experience on a small engineering team.
• Experience applying AI or LLM tooling to proprietary engineering data or desktop engineering software, with background in mechanical engineering, CAD, CAE, PLM, or a related domain.
• Familiarity with desktop automation and enterprise workstation constraints is valuable.

Compensation & Benefits

Salary range: $160,000 to $250,000 USD annually. Visa sponsorship is not available.

Location

On-site in San Francisco, California, United States.