Key skills
PythonPyTorchTensorFlowNumPyPandasscikit-learnLangChainADKAWSAzureGCPTerraformCI/CD pipelinesGitHub ActionsMachine LearningMLOpsRESTful APIsMicroservices ArchitectureAgentic AILarge Language ModelsInfrastructure-as-CodeAgile MethodologiesContinuous ImprovementAIMLLLMAgenticRESTfulGitHubAgileCI/CD
About this role
WEX is transforming how businesses operate by embedding advanced AI into their global payments and mobility platforms. The Sr. Staff AI/ML Engineer will lead the design and deployment of machine learning systems and collaborate with cross-functional teams to integrate AI capabilities into enterprise systems.
Responsibilities:
- Lead the design, implementation, and production deployment of machine learning and AI-driven systems—including LLM-based and agentic applications
- Partner with AI platform and product engineering teams to integrate advanced AI capabilities into WEX’s enterprise systems
- Design and maintain ML pipelines, from data ingestion to model deployment, ensuring scalability, observability, and reusability across teams
- Build and expose AI functionality via RESTful APIs and micro-services architectures
- Champion engineering best practices: CI/CD, infrastructure-as-code, testing automation, and continuous improvement
- Contribute to architectural decisions with a focus on security, compliance, and performance—especially in regulated industries such as payments and healthcare
- Collaborate cross-functionally with data scientists, ML engineers, and business stakeholders to align technical solutions with strategic goals
Requirements:
- 12+ years of professional software or ML engineering experience, with a track record of deploying production-grade AI systems
- Proficiency in Python and key machine learning frameworks (PyTorch, TensorFlow, or similar)
- Strong working knowledge of core libraries (NumPy, Pandas, scikit-learn) and LLM development frameworks (LangChain, ADK, or similar)
- Experience with cloud platforms (AWS preferred; Azure or GCP also valuable) and Infrastructure-as-Code tools like Terraform
- Deep familiarity with CI/CD pipelines and DevOps practices using GitHub Actions or similar platforms
- Demonstrated ability to operate in agile, collaborative, high-trust teams
- Bachelor's degree in Computer Science, Engineering, or a related discipline (Master's preferred)
- Experience in financial systems, data compliance, or building multi-tenant Agentic AI applications