WEX

Staff AI Engineer (L4)

WEX
Apply
3 days ago
Remote, United StatesStaff+
H1B sponsor

Base Salary

$185k - $217k/yr

Responsibilities

  • Partner with stakeholders to translate customer challenges into technical designs and AI strategies.
  • Architect, build, and deploy scalable, reliable, secure AI/ML platforms and tools.
  • Lead development, evaluation, fine-tuning, and monitoring of production AI/ML models, including LLMs and RAG architectures.
  • Design and standardize CI/CD pipelines using GitHub Actions and Infrastructure as Code using Terraform.
  • Monitor system health, lead critical incident response, and refactor architectures to remove bottlenecks.
  • Mentor L1/L2 engineers, conduct code reviews, and promote engineering best practices.
  • Evaluate emerging AI technologies and vendors and prototype new approaches.

Requirements

  • 8+ years of experience as an AI/ML Engineer, Data Scientist, or Software Engineer focused heavily on production ML systems.
  • Bachelor’s degree in Computer Science, Software Engineering, or a related field; a Master’s or PhD in AI/ML is preferred.
  • Production experience taking AI solutions from concept to high-scale production using Python, Java, Go, or C#.
  • Deep experience with generative AI, prompting strategies, RAG, fine-tuning, and agentic workflows.
  • Ability to design distributed systems and microservices architectures for model serving.
  • Expert proficiency in Python or other core languages including Java, Go, or C#.
  • Strong knowledge of PyTorch, TensorFlow, Scikit-learn, LangChain, LlamaIndex, and vLLM.
  • Advanced understanding of MLOps, feature stores, and model monitoring.
  • Hands-on Kubernetes cluster deployment and management experience and cloud experience, preferably AWS.
  • Strong data engineering skills involving SQL, data lakes, and processing pipelines.
  • Strong probability, statistics, and linear algebra knowledge.
  • Ability to lead technical discussions with non-technical stakeholders and drive consensus.
  • Financial or fintech industry experience is preferred.

Benefits

  • Remote work is available, but employees must reside within 30 miles of Portland, ME; Boston, MA; Chicago, IL; Dallas, TX; San Jose, CA; or Seattle, WA.
  • Benefits include health, dental, and vision insurance, retirement savings, paid time off, health savings and flexible spending accounts, life and disability insurance, tuition reimbursement, and additional benefits.
  • Non-sales roles are typically eligible for a quarterly or annual bonus under the applicable plan.

Tech Stack

AWSC#GitHub ActionsGoJavaKubernetesPythonPyTorchscikit-learnSQLTensorFlowTerraform
WEX

About WEX

5,001-10,000 employees

WEX is a public financial technology company that provides corporate payment solutions to fleets, healthcare/benefits administrators, and travel businesses. Its products include fuel and mobility cards, virtual card payments, and platforms for HSAs/FSAs/HRAs, with revenue from payment processing and software/service fees. Founded in 1983 and headquartered in Portland, Maine, WEX trades on the NYSE under WEX and was formerly known as Wright Express.

Contact me