Senior Staff AI/ML Engineer

WEX

. Lead the design, implementation, and production deployment of machine learning and AI-driven systems—including LLM-based and agentic applications .

Posted 5/23/2026full-timePortland • California, Illinois, Maine, Massachusetts, Washington • 🇺🇸 United StatesSenior💰 $220,000 - $255,800 per yearWebsite

Tech Stack

Tools & technologies

AWSAzureCloudGoogle Cloud PlatformNumpyPandasPythonPyTorchScikit-LearnTensorflow

About the role

Key responsibilities & impact

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

What you’ll need

12+ years of professional software or ML engineering experience
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)
Deep familiarity with CI/CD pipelines and DevOps practices using GitHub Actions or similar platforms
Bachelor’s degree in Computer Science, Engineering, or a related discipline (Master’s preferred)

Benefits

Comp & perks

health, dental and vision insurances
retirement savings plan
paid time off
health savings account
flexible spending accounts
life insurance
disability insurance
tuition reimbursement

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

machine learningAI-driven systemsPythonPyTorchTensorFlowNumPyPandasscikit-learnLLM development frameworksRESTful APIs

Soft Skills

leadershipcollaborationcommunicationstrategic alignmentcontinuous improvement

Certifications

Bachelor’s degree in Computer ScienceBachelor’s degree in EngineeringMaster’s degree in related discipline