AI Principal Engineer

WEX

Principal AI/ML Engineer at WEX responsible for designing the cloud-native AI/ML engineering platform. Collaborating with cross-functional teams to drive technology development and architecture.

Posted 5/27/2026full-timeRemote • California, Illinois, Maine, Massachusetts, Oregon • 🇺🇸 United StatesLead💰 $250,300 - $289,000 per yearWebsite

Tech Stack

Tools & technologies

CloudJavaKubernetesPythonRaySDLCSparkTerraform

About the role

Key responsibilities & impact

Lead and drive the development of technology and platform for the company's AI/ML engineering needs
Help design the architecture and lead the implementation of the AI/ML infrastructure, platform and services
Challenge the status quo and drive for better technical solutions
Guide your team to leverage data and/or AI technology or tools in your design and development
Show strong thought leadership and influence peers
Partner closely with other senior engineers to understand business needs
Engage with external partners to evaluate technologies and services that complement AI/ML strategy
Create, prioritize, manage, and execute roadmaps and project plans
Drive technical vision and strategy of AI/ML infrastructure
Provide technical guidance and effective training & mentorship to engineering teams

Requirements

What you’ll need

15-20 years of software design and development experience at a large scale
Strong software development skills in programming languages, willing to learn Java and Python
Good experience in building machine learning supervised and unsupervised models
Experience in Reinforcement Learning is a plus
Experience in using LLM models a plus
Broad understanding of distributed computing frameworks like Ray and Spark
Demonstrable experience in Kubernetes and Terraform
Experience with Sagemaker a plus
Broad understanding of cloud architecture tools and services, such as S3, EMR, Kubernetes, Lambda functions
Extensive experience in full software development life cycle E2E including in production monitoring
Rich experience and deep expertise in the areas of Big Data and large scale data platforms, especially in AI and ML
Highly experienced and deep understanding of CI/CD automation
Ability to encourage and foster a culture of visibility and transparency across teams
Bachelor degree in Computer Science or a related field, higher degree is a plus

Benefits

Comp & perks

Health, dental, and vision insurances
Retirement savings plan
Paid time off
Health savings account
Flexible spending accounts
Life insurance
Disability insurance
Tuition reimbursement

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

software designsoftware developmentJavaPythonmachine learningReinforcement LearningLLM modelsdistributed computingKubernetesTerraform

Soft Skills

thought leadershipinfluencementorshipteam guidanceproject managementcommunicationcollaborationtransparencyproblem-solvinginnovation

Certifications

Bachelor degree in Computer Sciencehigher degree in related field