Full-Time

Principal AI/ML Engineer

Confirmed live in the last 24 hours

Wex

Wex

1,001-5,000 employees

B2B services for benefits and payments

Compensation Overview

$203k - $270k/yr

+ Bonus

Expert

Boston, MA, USA + 3 more

More locations: Oakland, CA, USA | Chicago, IL, USA | Portland, ME, USA

Candidates can be based in any of the listed cities: Bay Area, CA; Portland, ME; Chicago, IL; Boston, MA. The job is likely hybrid, requiring some in-office presence depending on the location.

Category
Applied Machine Learning
AI Research
AI & Machine Learning
Required Skills
LLM
Kubernetes
Microsoft Azure
Python
Data Science
Apache Spark
Machine Learning
Java
AWS
Terraform
Reinforcement Learning
Requirements
  • 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.
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.
Desired Qualifications
  • Experience in Reinforcement Learning is a plus.
  • Experience in using LLM models a plus.
  • Experience with Sagemaker a plus.
  • An understanding of and experience in payment organizations is a plus.

WEX Inc. is a global commerce platform that helps businesses manage employee benefits, fleet operations, and business payments. It offers solutions like health savings accounts, fuel cards, and automated payment processes to streamline operations and improve efficiency. WEX stands out from competitors by providing a comprehensive suite of specialized services tailored to various business needs. The company aims to reduce operational costs for businesses while promoting diversity and corporate social responsibility.

Company Size

1,001-5,000

Company Stage

IPO

Headquarters

Portland, Oregon

Founded

1983

Simplify Jobs

Simplify's Take

What believers are saying

  • Virtual card adoption is rising, with 90% of firms pursuing their use.
  • WEX's solutions improve operational efficiency, helping businesses focus on core activities.
  • The partnership with Sabre enhances WEX's reach in the travel industry, offering faster payments.

What critics are saying

  • Slow supplier adoption of virtual cards may hinder WEX's growth in business payments.
  • Legacy system complexities could impact WEX's ability to offer cutting-edge solutions.
  • Trust issues with agentic AI could affect WEX's B2B operations, built on human accountability.

What makes Wex unique

  • WEX offers comprehensive B2B solutions in employee benefits, fleet management, and business payments.
  • The company provides virtual card technology, enhancing security and cash flow for businesses.
  • WEX's partnership with Sabre simplifies global travel transactions, reducing fraud and conversion costs.

Help us improve and share your feedback! Did you find this helpful?

Benefits

Health Insurance

Dental Insurance

Vision Insurance

Life Insurance

Disability Insurance

Health Savings Account/Flexible Spending Account

Paid Vacation

Tuition Reimbursement

Company News

PYMNTS
Jun 12th, 2025
Supplier Enablement Programs Accelerate Virtual Card Use

The B2B payments landscape has long been characterized by competing interests, with buyers seeking to extend payment terms while suppliers push for prompt remittance.In joint research from WEX and PYMNTS Intelligence as part of the continuing Business Tracker series, virtual cards present a compelling solution that offers significant advantages for both accounts payable (AP) and accounts receivable (AR) departments, yet supplier adoption faces notable obstacles that require strategic intervention. Gaining Traction. Virtual card payments are gaining traction, with nearly 9 in 10 firms adopting or pursuing their use for business payments, the data show.While buyers benefit from enhanced security, improved spend management, and potential rebates, suppliers stand to gain significantly from faster payments and improved cash flow

PYMNTS
Jun 11th, 2025
Cloud Giants Hit Slow Lane As Legacy Systems Stall Enterprise Upgrades

For years, digital transformation took the lion’s share of investment, as companies wanted to move fast and capture new efficiencies to compete with even faster-moving, more operationally streamlined firms. Today, however, enterprise digital transformation at the highest level is no longer just about buying modern tools and leaning into cloud migrations. For some of the world’s biggest and most sprawling businesses, it can be about timing, culture and a hands-on approach to unraveling the complexities of long-standing legacy back offices. When cloud-native companies like Snowflake and Databricks first rocketed to prominence, their success was buoyed by the growth of digital-first businesses. These organizations, often startups or cloud-native unicorns, needed scalable, agile infrastructure to fuel growth. Cloud data warehouses and platforms were the answer, offering a way to store, manage and analyze data without the burdens of traditional IT infrastructure

PYMNTS
Jun 10th, 2025
Agentic Ai Needs To Solve For Trust Before Conquering B2B

The B2B space barely had time to wrap its head around generative artificial intelligence (GenAI) before the next iteration, agentic AI, arrived. And if GenAI jolted the B2B world with its ability to produce language, content and code at scale, agentic AI ups the ante.These new systems are not just capable of completing tasks; they’re able to take initiative. From autonomously conducting sales outreach to managing procurement and even inching toward handling payment operations, agentic AI offers a leaner, faster, smarter B2B machine. But there’s a paradox here that businesses are only beginning to grapple with: B2B runs on trust. And trust doesn’t scale easily. While consumer markets might tolerate a bit of algorithmic overreach or a chatbot gone rogue, B2B relationships are often multimillion-dollar engagements built on handshakes, track records and human accountability

PYMNTS
Jun 9th, 2025
Wex And Sabre Partner On Virtual Cards For Travel Agencies

In this strategic global partnership, the customers of Sabre Direct Pay, which is Sabre’s dedicated travel payments hub, will gain access to WEX’s virtual card technology, the companies said in a Monday (June 9) press release. This combination will simplify transactions across multiple currencies, borders and time zones, according to the release. Virtual card solutions will provide travel intermediaries with faster payments; a way to avoid foreign exchange and conversion expenses; rich metadata with each transaction; a simpler chargeback process; and decreased vulnerability to fraud, per the release

PYMNTS
Jun 2nd, 2025
Bringing The Momentum Back To Enterprise Digital Transformation

Digital transformation should result in digital capabilities, but it appears to be losing its momentum in many organizations. As companies grapple with tighter budgets and shorter executive tenures, many flashy tech initiatives have failed to evolve into lasting business capabilities — prompting a rethink focused on speed, impact and alignment with core enterprise goals. Whether [] The post Bringing the Momentum Back to Enterprise Digital Transformation appeared first on PYMNTS.com.