Note: The job is a remote job and is open to candidates in USA. WEX Inc is a company focused on providing innovative technology solutions, and they are seeking a Senior Software AI Engineer to build high-performance software for their enterprise AI. This role involves developing infrastructure for next-generation AI workflows, contributing to AI infrastructure, and maintaining high standards for code quality.
Responsibilities
- AI Infrastructure Implementation: Contribute to the development of a high-scale, AI-ready Data Lakehouse optimized for AI Agent operations and low-latency context retrieval
- Agentic R&D & Prototyping: Hands-on prototyping of emerging architectural patterns, such as Multi-Agent Orchestration and autonomous long-term memory management
- Engineering Excellence: Maintain high standards for code quality and CI/CD, participating in cross-functional architecture reviews and troubleshooting complex system bottlenecks
- Agentic Ecosystem Development: Build platform-level interfaces for agentic workflows, focusing on "Host-to-Server" communication and tool-execution environments
- Contextual Fabric Construction: Develop systems that move beyond basic search into Reasoning-based Retrieval, helping the platform understand the intent behind an agent's query
- Protocol Integration: Implement emerging standards like the Model Context Protocol (MCP) and Agentic RAG to ensure interoperability between the platform and various LLM providers
Skills
- 6+ years of software engineering experience with a focus on distributed systems
- Proficiency in Java or Scala and Python
- Experience building extensible APIs and libraries used by other developers
- A preference for building automated, software-defined infrastructure over manual configuration
- Hands-on experience with agent development frameworks such as LangGraph 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 the CI/CD pipeline
- Understanding of Human-in-the-Loop (HITL) triggers to ensure safety in autonomous systems
- Experience with GitOps workflows (e.g., ArgoCD or Flux) to manage versioned platform configurations and AI prompt templates
- Proficiency in Terraform to build reusable modules that enforce organizational standards across cloud accounts
- Ability to design CI/CD pipelines (e.g., GitHub Actions) that integrate automated testing and security scanning
- Hands-on experience navigating and configuring AWS and Azure Management Consoles, including core services like IAM, EC2/VMs, and S3/Blob storage
- Experience using OpenTelemetry (OTel) to track system performance and AI-specific success metrics
- A proven track record of collaborating across autonomous teams to drive the adoption of new technologies
- Ability to clearly communicate technical trade-offs to both fellow engineers and stakeholders
- Bachelor's or Master's degree in Computer Science (Distributed Systems focus) preferred, or equivalent deep industry experience
Benefits
- Health, dental and vision insurances
- Retirement savings plan
- Paid time off
- Health savings account
- Flexible spending accounts
- Life insurance
- Disability insurance
- Tuition reimbursement
- And more
Company Overview
Company H1B Sponsorship