The Flex

The Flex

Staff Software Engineer, AI Platform

USA · Staff+ · Full-time

Sponsorship not specified$200k-$250kDetected 5 days ago
TypeScriptPythonJavaGoDistributed SystemsSQLNoSQLPostgreSQLMySQLVector DatabasesAWSTerraformLLMsRAGAgentic AILangGraphAI OrchestrationLeadershipCommunicationMentoring

About the role

  • As a Staff AI Engineer at Flex, you'll help set technical direction and shape the long-term architecture of the AI systems and tools that all departments use to power their agentic workflows.
  • You'll also play a critical role in mentoring, elevating technical standards, and ensuring the success of our Flex's transformation into an AI-first organization.

Responsibilities

  • Design, implement, and maintain the agent platform including orchestration, tool execution, sandboxed code runtimes, observability, and the developer-facing SDKs and UI.
  • Design and develop new APIs, SDKs, and developer surfaces that let other teams onboard agents quickly and safely, with strong defaults for evals, guardrails, human-in-the-loop review, and audit.
  • Scale the platform to support a growing catalog of agents, skills, and MCP integrations across Flex while maintaining performance, reliability, and cost efficiency.
  • Continuously improve our daily operations with automation, tooling, design evolution, and observability, including agent-trace visibility, evaluation pipelines, and sandbox runtime monitoring.
  • Set the bar for AI-assisted engineering practices and responsible AI use across Flex, mentor engineers on how to design, build, and operate agentic systems, and bring outside innovation back into our roadmap.
  • Experience building AI-first workflows end-to-end, including LLM application design, prompt and context engineering, RAG, and human-in-the-loop patterns.
  • We are growing quickly, but deliberately, with a focus on building an inclusive culture.

Requirements

  • Strong communication skills with the ability to understand and explain technical issues to a non-technical audience.
  • Skills that are required:
  • 6+ years of experience in software development or a related field, with at least 2+ years building production AI or agentic systems.
  • Hands-on experience with coding CLIs and AI dev tools (Claude Code, Codex, OpenCode, Cursor, or equivalents) - both as a daily-driver user and as a builder integrating them into team workflows.
  • Experience with agent infrastructure and code-execution sandboxes (Modal, E2B, Cloudflare Sandboxes, Daytona, or Firecracker/gVisor-based runtimes), with a clear point of view on isolation, scaling, and cost trade-offs.

Nice to have

  • experience with model and prompt evaluation tooling, secure tool-use sandboxing, retrieval and vector stores, and operating multi-tenant AI platforms at scale.
  • The successful candidate's starting pay will be commensurate with their experience, qualifications, and Flex's internal leveling guidelines and benchmarks.

Skills

  • Experience in technical leadership, guiding engineering teams through complex projects.
  • Experience with Java, Python, Golang, or TypeScript/Node.js.
  • Strong experience with relational SQL and NoSQL databases, such as MySQL/PostgreSQL/DynamoDB.

Compensation

  • Flex takes a market-based approach to pay, and compensation may vary depending on your primary work location.
  • The successful candidate's starting pay will be commensurate with their experience, qualifications, and Flex's internal leveling guidelines and benchmarks.
  • $200,000-$250,000 USD
  • $180,000-$225,000 USD
  • $170,000-$212,000 USD

Benefits

  • Competitive medical, dental, and vision
  • Company equity
  • Unlimited paid time off + 13 company paid holidays
  • Parental leave

Company info

  • For full-time U.S. employees we offer:
  • For full-time non-U.S. employees, we offer:

Equal opportunity

  • Our dynamic team has incredible perspectives to share, just as we know you do, and we take great pride in being an equal opportunity workplace.
  • equal opportunity workplace.
Apply directly at The Flex →Create a free account for alerts like thisView The Flex immigration profile

This listing is sourced directly from The Flex's careers page and normalized into a canonical job model.