Senior Staff AI Engineer, Production ML & LLM

Maine Tech Hub; Portland (ME); On-site; USD 185,000 - 217,000; Full time

Job summary

WEX Inc. in South Portland, Maine seeks a Staff AI Engineer to lead production ML systems and shape AI strategy across platforms. You’ll mentor engineers, architect scalable solutions, and collaborate with product teams to turn challenges into reliable AI capabilities.

You will own end-to-end ML workflows, deploy via CI/CD, and work with cloud platforms and data pipelines to deliver high-impact AI features for customers and the business.

Qualifications

• 8+ years of experience in AI/ML/engineer roles focusing on production ML systems.
• Production experience delivering AI solutions at scale using Java, C#, Go, or Python.
• Experience with distributed systems, microservices, and model serving.
• Strong collaboration with stakeholders and leadership.

Responsibilities

• Technical Strategy & Architecture: translate customer challenges into robust AI designs and strategies.
• End-to-End Ownership: build and deploy high-performance AI/ML platforms with reliability and security.
• Advanced Model Development: lead development, evaluation, tuning, and monitoring of production models (LLMs, RAG).
• Operational Excellence: design CI/CD pipelines and IaC to automate deployments.
• Reliability & Scale: monitor systems, lead incident response, drive refactoring to remove bottlenecks.
• Mentorship: mentor junior engineers and conduct code reviews.
• Innovation: evaluate new AI tech and prototype new approaches.

Skills

AI/ML Engineer

Data Scientist

Software Engineer

Production ML systems

Python

Java

C#

Go

LLM

RAG

Distributed systems

Microservices

Kubernetes

AWS

CI/CD

Terraform

SQL

NoSQL

Education

Bachelor’s degree in Computer Science or related field

Master’s or PhD in AI/ML preferred

Tools

LangChain

LlamaIndex

vLLM

PyTorch

TensorFlow

Scikit-learn

GitHub Actions

Terraform

Kubernetes

AWS

Job description

WEX Inc. in South Portland, Maine seeks a Staff AI Engineer to lead production ML systems and shape AI strategy across platforms. You’ll mentor engineers, architect scalable solutions, and collaborate with product teams to turn challenges into reliable AI capabilities.

You will own end-to-end ML workflows, deploy via CI/CD, and work with cloud platforms and data pipelines to deliver high-impact AI features for customers and the business.