WEX

AI Principal Engineer

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

Base Salary

$250k - $289k/yr

Responsibilities

  • Lead development of WEX’s AI/ML engineering technology and platform, ensuring reliability, performance, flexibility, and functional richness.
  • Design the architecture and lead implementation of AI/ML infrastructure, platforms, applications, and services.
  • Drive the technical vision, strategy, roadmaps, project plans, and commitments for AI/ML infrastructure.
  • Partner with product, engineering, data science, security, SRE, architecture, and business teams on integrations and technical solutions.
  • Ensure data security and privacy through appropriate encryption and access controls.
  • Evaluate external technologies and vendors that support WEX’s AI/ML strategy.
  • Establish quality goals, monitor development and operational performance, and uphold security requirements, SLOs, error budgets, DORA metrics, and OKRs.
  • Provide technical guidance, mentorship, training, and strategic leadership to engineers and engineering leaders.
  • Advise senior leadership on complex technical and business initiatives and lead teams through execution.

Requirements

  • 15–20 years of software design and development experience at large scale.
  • Strong software development skills and willingness to program in Java and Python.
  • Experience building supervised and unsupervised machine learning models; reinforcement learning experience is preferred.
  • Experience with distributed computing frameworks such as Ray and Spark.
  • Demonstrable experience with Kubernetes and Terraform.
  • Experience with cloud architecture and services including S3, EMR, Kubernetes, and Lambda; AWS and Azure experience is highly desirable.
  • Experience with SageMaker and LLM models is preferred.
  • Hands-on experience across architecture, development, deployment, production operations, monitoring, automation, quality, and the full software development lifecycle.
  • Deep experience with big data, large-scale data platforms, AI, and ML.
  • Strong experience with CI/CD automation and leading engineering teams through modern development practices.
  • Excellent analytical, problem-solving, communication, stakeholder-management, and mentoring skills.
  • Payment-organization experience is preferred.
  • Bachelor’s degree in computer science or a related field; a higher related degree is preferred.

Benefits

  • Health, dental, and vision insurance.
  • Retirement savings plan, paid time off, health savings account, flexible spending accounts, life insurance, and disability insurance.
  • Tuition reimbursement and additional benefits.
  • The role can be based in San Francisco Bay Area, Chicago, Portland, Maine, or Boston.
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