Staff AI Engineer (Remote) - Lead Scalable AI Platforms

WEX, Inc.; United States; On-site; USD 185,000 - 217,000; Full time

Job summary

WEX, Inc. is seeking a Staff AI Engineer (L4) to lead the design and delivery of production AI platforms. Remote work within specified US locations; you will mentor engineers, set technical standards, and partner with product leadership to translate business goals into scalable AI solutions.

You will own end-to-end AI systems—from model development and evaluation to deployment, monitoring, and ongoing optimization—driving reliability and impact across the business.

Qualifications

• 8+ years building production ML systems and AI platforms.
• Strong background in distributed systems and scalable architectures.
• Experience leading engineers and collaborating with product teams.

Responsibilities

• Define AI strategy and architecture with stakeholders.
• Lead end-to-end development of production ML platforms.
• Mentor junior engineers and enforce engineering standards.

Skills

Distributed Systems

Software Architecture

Technical Leadership

Python

LLM/Generative AI

Education

Bachelor's degree in CS/SE

Tools

LangChain

LlamaIndex

vLLM

Kubernetes

Terraform

AWS

PyTorch

TensorFlow

Job description

WEX, Inc. is seeking a Staff AI Engineer (L4) to lead the design and delivery of production AI platforms. Remote work within specified US locations; you will mentor engineers, set technical standards, and partner with product leadership to translate business goals into scalable AI solutions.

You will own end-to-end AI systems—from model development and evaluation to deployment, monitoring, and ongoing optimization—driving reliability and impact across the business.