Sr. Staff AI/ML Engineer

Role Description

This is a remote position; however, the candidate must reside within 30 miles of one of the following locations: Portland, ME; Boston, MA; Chicago, IL; Dallas, TX; San Francisco Bay Area, CA; and Seattle/WA.

What You’ll Do:

• 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.

Qualifications

• 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).
• Bonus: Experience in financial systems, data compliance, or building multi-tenant Agentic AI applications.

Benefits

• Health, dental and vision insurances.
• Retirement savings plan.
• Paid time off.
• Health savings account.
• Flexible spending accounts.
• Life insurance.
• Disability insurance.
• Tuition reimbursement.
• Quarterly or annual bonus based on role and applicable plan.

Pay Range

$220,000.00 - $255,800.00