Staff Software Engineer, AI Platform

Responsibilities

• Design, implement, and maintain the agent platform including orchestration, tool execution, and sandboxed code runtimes
• Develop APIs, SDKs, and developer surfaces for agent onboarding
• Scale the platform to support a growing catalog of agents, skills, and MCP integrations
• Improve daily operations with automation, observability, and evaluation pipelines
• Partner with cross-functional teams to identify and implement high-leverage AI use cases
• Mentor engineers and set technical standards for responsible AI use

Requirements

• 6+ years of software development experience
• 2+ years building production AI or agentic systems
• Experience with technical leadership and guiding engineering teams
• Hands-on experience with AI dev tools (Claude Code, Cursor, etc.)
• Experience with agent infrastructure and code-execution sandboxes (Modal, E2B, etc.)
• Proficiency in Java, Python, Golang, or TypeScript/Node.js
• Strong experience with relational SQL and NoSQL databases
• Experience with AWS cloud services (ECS/EKS, Lambda, API Gateway)
• Experience with Infrastructure as Code (Terraform, AWS CDK)

Preferred Qualifications

• Experience with model and prompt evaluation tooling
• Experience with secure tool-use sandboxing
• Experience with retrieval and vector stores
• Experience operating multi-tenant AI platforms at scale

Benefits

• Competitive medical, dental, and vision
• Company equity
• 401(k) plan with company match
• Unlimited paid time off + 13 company paid holidays
• Parental leave
• Free Flex subscription

About the Company

Flex is a growth-stage FinTech company creating a better rent payment experience by enabling users to pay rent on a schedule that fits their finances.