Demo

Principal AI/ML Engineer

WEX
Chicago, IL Full Time
POSTED ON 1/9/2026
AVAILABLE BEFORE 6/7/2026
Responsibilities

  • Lead and drive the development of technology and platform for the company's AI/ML engineering needs, ensure the functional richness, reliability, performance, and flexibility of this platform
  • Help design the architecture and lead the implementation of the AI/ML infrastructure, platform and services.
  • Challenge the status quo and hold a high bar to drive for better and better technical solutions to effectively serve all the needs in the broad risk area of Wex.
  • Guide your team to leverage data and/or AI technology or tools in your design and development for high productivity and better solution quality. Show strong thought leadership and influence your peers in this area.
  • Closely partner with the other senior engineers across the organizations at Wex to deeply understand the business needs and jointly come up with technical solutions to serve these needs effectively. Demonstrate strong thought leadership and partnership.
  • Partner closely with information security teams to ensure the security and privacy of data, implementing encryption and access controls as required.
  • Partner with solutions architecture and LOB product and engineering orgs on integrations of the AI/ML Platform with their products/systems, and ensure reliable and performed execution.
  • Engage with external partners and vendors to evaluate technologies and services that complement the company’s AI/ML strategy.
  • Create, prioritize, communicate, manage, and execute roadmaps, project plans, and commitments.
  • Report on the status of development, quality, operations, and system performance to management and partners.
  • Drive the technical vision and strategy of AI/ML infrastructure to serve the business needs for the near term and long term. A close partnership with the Data Science and ML Engineering teams and the operational reporting teams of applications
  • Set a high bar, and provide technical guidance and effective training & mentorship to the engineering teams in your group, including engineering principles, skills, and best practices. Uphold strong DORA metrics across the org. Foster an environment of continuous learning and innovation. Build an inclusive, productive, innovative, and engaging Risk group, including attracting high talents from the industry. Guide the leaders in your group to do the same.
  • Set clear, measurable quality goals for your group in a data-driven way
  • Partner with the security org and SRE org to ensure continuous improvement and meeting of security requirements, SLOs, and error budget.
  • Take strong financial responsibility for your team.
  • Hold yourself and your team accountable for delivering quality results using defined OKRs.
  • Advise Senior Leadership team on highly complex situations that impact our organization and then lead teams through the initiative, driving excellent quality results and achieving defined OKRs across the organization.
  • Provide strategic thought leadership on business initiatives using your deep technical and business industry expertise.

Qualifications & Experience

  • A technically deep and innovative Software Engineer leader able to act and deliver to the business needs.
  • Have 15-20 years of software design and development experience at a large scale.
  • Strong software development skills in his or her chosen programming language, but ready and willing to learn and program in 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. Experience in AWS and Azure is highly desirable.
  • Deep technically; hands-on in architecture design, development, deployment, and production operation; familiar with agile software development and modern development tools and frameworks; holding strong engineering principles including automation and quality as well as best practices with a high bar. 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.
  • Rich experience in combined engineering practice and agile development. Experienced in leading and guiding teams for effective development using this approach.
  • Excellent analytical skills; Capability in identifying and resolving complex problems.
  • An understanding of and experience in payment organizations is a plus.
  • A mentor to individual contributors of different seniority levels and expertise.
  • Ability to encourage and foster a culture of visibility and transparency across teams
  • Strong written and verbal communication skills
  • Strong philosophy of continuous improvement
  • Bachelor degree in Computer Science or a related field. A higher degree in a related field is a plus.

Leadership & Personal Characteristics

Beyond Experience, The Right Technical Leadership Competencies And Personal Style Are Critical To Success As The Principal AI/ML Engineer. The Candidate Will Model WEX's Commitment To Innovation, Integrity, Execution, Relationships, Community, And The Excellence That They Commit To Their Customers. They Will Have a Strong Corporate Stewardship Philosophy And Leadership Style Combined With The Following

  • Demonstrates a collaborative, engaging, mission-driven style with good listening skills combined with gravitas and self-confidence in order to effectively interact with senior leadership and other key constituencies with humility, empathy, transparency, cultural sensitivity, and agility.
  • Champions a culture of high standards, pushes for progress, embraces change and challenges the status quo based on integrity, strong partnership and financial responsibility. Possess uncompromising ethics and unwavering integrity.
  • Able to operate in a fast-paced entrepreneurial environment with a hands-on approach.
  • Communicates effectively in small and large group formats. Able to articulate a compelling vision and purpose that excites the team and drives outstanding performance and service delivery excellence.
  • Possesses a high level of intellectual curiosity and learning agility. A natural change agent with a combination of vision and execution skills.
  • Understands nuance and how to drive change through influence across a complex business. Superb stakeholder management and comfortable operating in a fast-changing environment.

The base pay range represents the anticipated low and high end of the pay range for this position. Actual pay rates will vary and will be based on various factors, such as your qualifications, skills, competencies, and proficiency for the role. Base pay is one component of WEX's total compensation package. Most sales positions are eligible for commission under the terms of an applicable plan. Non-sales roles are typically eligible for a quarterly or annual bonus based on their role and applicable plan. WEX's comprehensive and market competitive benefits are designed to support your personal and professional well-being. Benefits include health, dental and vision insurances, retirement savings plan, paid time off, health savings account, flexible spending accounts, life insurance, disability insurance, tuition reimbursement, and more. For more information, check out the "About Us" section.

Pay Range: $203,000.00 - $270,000.00

Salary : $203,000 - $270,000

If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

What is the career path for a Principal AI/ML Engineer?

Sign up to receive alerts about other jobs on the Principal AI/ML Engineer career path by checking the boxes next to the positions that interest you.
Income Estimation: 
$131,953 - $159,624
Income Estimation: 
$169,825 - $204,021
Income Estimation: 
$166,631 - $195,636
Income Estimation: 
$162,237 - $199,353
Income Estimation: 
$181,083 - $218,117
Income Estimation: 
$97,257 - $120,701
Income Estimation: 
$123,167 - $152,295
Income Estimation: 
$123,167 - $152,295
Income Estimation: 
$146,673 - $180,130
Income Estimation: 
$146,673 - $180,130
Income Estimation: 
$176,149 - $220,529
Income Estimation: 
$176,149 - $220,529
Income Estimation: 
$156,679 - $196,968
View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

Job openings at WEX

  • WEX Portland, ME
  • This is a remote position, however, the candidate must reside within 30 miles of one of the following locations: Portland ME, Chicago IL, Boston MA, Washin... more
  • 12 Days Ago

  • WEX Seattle, WA
  • We are seeking a seasoned Sr. Manager of Engineering in the WEX Mobility Engineering organization . This role will lead the engineering team for our Over T... more
  • 13 Days Ago

  • WEX San Francisco, CA
  • This is a remote position; however, the candidate must reside within 30 miles of one of the following locations: Portland, ME; Boston, MA; Chicago, IL; San... more
  • 13 Days Ago

  • WEX Boston, MA
  • This is a remote position; however, the candidate must reside within 30 miles of one of the following locations: Boston, MA; San Francisco Bay Area, CA; Da... more
  • 14 Days Ago


Not the job you're looking for? Here are some other Principal AI/ML Engineer jobs in the Chicago, IL area that may be a better fit.

  • Chamberlain Advisors Deerfield, IL
  • Title: Data Engineer – AI / ML Location: Hybrid, 3 days/week – Deerfield, IL Duration & Type: Initial 6-Month Contract Compensation: Competitive W2 Hourly ... more
  • 13 Days Ago

  • GNRSystems Chicago, IL
  • Job Details We are looking for an experienced AI / Machine Learning Engineer to join our team in Chicago, IL (Hybrid) . The candidate will be responsible f... more
  • 27 Days Ago

AI Assistant is available now!

Feel free to start your new journey!