Sr Software AI Engineer - Context Engineering

As an AI Platform Engineer (SDE3), you will be a key builder of the high-performance software foundation that powers our enterprise AI. While your expertise lies in distributed systems and cloud-native architecture, you will apply these skills specifically to the "Context Layer"—the specialized infrastructure required to fuel next-generation Agentic AI workflows. You will work at the intersection of systems programming and modern AI infrastructure to solve practical problems in real-time data orchestration and multi-cloud compute optimization.

This is a "platform-as-a-product" role where you build the tools and SDKs that enable other engineers to build autonomous agents with ease.

• Contribute to the development of a high‑scale, AI‑ready Data Lakehouse optimized for AI Agent operations and low‑latency context retrieval.
• Hands‑on prototyping of emerging architectural patterns, such as Multi‑Agent Orchestration and autonomous long‑term memory management.
• Maintain high standards for code quality and CI/CD, participating in cross-functional architecture reviews and troubleshooting complex system bottlenecks.
• Build platform‑level interfaces for agentic workflows, focusing on "Host-to-Server" communication and tool‑execution environments.
• Develop systems that move beyond basic search into Reasoning‑based Retrieval, helping the platform understand the intent behind an agent's query.
• Implement emerging standards like the Model Context Protocol (MCP) and Agentic RAG to ensure interoperability between the platform and various LLM providers.

• 6+ years of software engineering experience with a focus on distributed systems.
• Proficiency in Java, Scala, or Python.
• Experience building extensible APIs and libraries used by other developers.
• Preference for building automated, software‑defined infrastructure over manual configuration.
• Hands‑on experience with agent development frameworks such as Lang Graph or CrewAI and the transition from static RAG to Agentic RAG.
• Knowledge of the Model Context Protocol (MCP) and how it allows AI agents to interact with diverse data sources.
• Experience building AI‑native features, including automated LLM‑based evaluations within CI/CD pipelines.
• Understanding of Human-in-the-Loop (HITL) triggers to ensure safety in autonomous systems.
• Experience with Git Ops workflows (e.g., ArgoCD or Flux) to manage versioned platform configurations and AI prompt templates.
• Proficiency in Terraform for building reusable modules that enforce organizational standards across cloud accounts.
• Ability to design CI/CD pipelines (e.g., Git Hub Actions) that integrate automated testing and security scanning.
• Hands‑on experience with AWS and Azure, including core services such as IAM, EC2/VMs, and S3/Blob storage.
• Experience using Open Telemetry to track system performance and AI‑specific success metrics.
• Strong collaboration skills across autonomous teams to drive the adoption of new technologies.
• Excellent communication skills to convey technical trade-offs to engineers and stakeholders.
• Education:
  
  Bachelor’s or Master’s degree in Computer Science (Distributed Systems focus) preferred, or equivalent deep industry experience.

Base pay range: $ – $. Compensation may include other components such as bonus and benefits. Benefits include health, dental, and vision insurance, retirement savings plan, paid time off, health savings account, flexible spending accounts, life insurance, disability insurance, tuition reimbursement, and more.

WEX is an equal opportunity employer committed to diversity and inclusion in the workplace. All qualified applicants will receive consideration for employment without regard to sex, race, color, age, national origin, religion, sexual orientation, gender identity, protected veteran status, disability or other protected status. WEX promotes a drug‑free workplace. Qualified individuals with a disability have the right to request a reasonable accommodation.

If you require a reasonable accommodation as a result of your disability at any point in the job application process, please submit your request through our Reasonable Accommodation Request Form.