AI Principal Engineer
AI summary of the role
This Principal AI/ML Engineer will own the design and implementation of WEX's cloud-native AI/ML engineering platform, supporting both GenAI and traditional AI/ML across multiple lines of business.
No longer in the current catalog. Last included 2026-09-09. Check the employer’s posting for availability.
What you’ll do
- Lead design and implementation of the cloud-native AI/ML engineering platform (GenAI and traditional AI/ML).
- Drive technical vision, architecture, and engineering best practices for AI/ML infrastructure.
- Partner with Data & Analytics, Risk & Compliance, LOB product/engineering, and security teams to align platform with business needs.
- Engage with external partners and vendors to evaluate complementary AI/ML technologies.
What you’ll bring
- 15-20 years of software design and development experience at large scale.
- Deep technical expertise in building and scaling data lake environments and AI/ML platforms.
- Strong programming skills in Java and Python.
- Experience with distributed computing frameworks like Ray and Spark.
Technologies
Java · Python · Ray · Spark · Kubernetes · Terraform · Sagemaker · AWS · Azure · S3 · EMR · Lambda
About Wex
WEX runs a global commerce and payments platform across three segments — fleet/mobility fuel cards, corporate B2B/virtual-card payments, and employee-benefits (HSA/FSA/COBRA) administration.
Public · 5000+ people
Source and classification
Internal deployment & tooling · Evidence for this classification:
Position Summary As a Principal AI/ML engineer, this technical leader will be responsible for the design and implementation of the cloud-native AI/ML engineering platform (supporting both GenAI and traditional AI/ML technology and applications). This technologist will drive the technology development and architecture in these areas to ensure the platforms, systems, tools, models, and services generated from this team are technically sound with strong security, quality, reliability, usability, scalability, performance, efficiency, and operability to serve the fast growing needs of Wex and customers. It’s important to balance the near term and long term needs. To do these, you will be closely partnering with other leaders and teams in the Tech org, Digital org, and other orgs in the company including Data & Analytics, Risk & Compliance, the Line of Business (LOB) 3 legs (commercial,
More from the job description
Position Summary As a Principal AI/ML engineer, this technical leader will be responsible for the design and implementation of the cloud-native AI/ML engineering platform (supporting both GenAI and traditional AI/ML technology and applications). This technologist will drive the technology development and architecture in these areas to ensure the platforms, systems, tools, models, and services generated from this team are technically sound with strong security, quality, reliability, usability, scalability, performance, efficiency, and operability to serve the fast growing needs of Wex and customers. It’s important to balance the near term and long term needs. To do these, you will be closely partnering with other leaders and teams in the Tech org, Digital org, and other orgs in the company including Data & Analytics, Risk & Compliance, the Line of Business (LOB) 3 legs (commercial, digital, and technology), Security, central architecture & PaaS team, Global Tech Services, Finance, Marketing, and customer service orgs to understand their business and needs for Data Lake technologies, platforms, services, and guide our design, development, and productionalization accordingly. This is a fantastic opportunity for a highly motivated and passionate AI/ML engineer to apply deep expertise and rich experience in AI Engineering to develop technology, platforms and services, and apply th [... source excerpt omitted ...] ness with a big customer base to generate real business impact in a fast growing commerce platform and payment tech company during its technology transformation stage to meet high customer demands. At the same time, you will learn and grow your career by solving large scale and complex problems. This Principal AI/ML Engineer will have technical ownership of WEX’s AI Engineering implementation, including but not limited to architecture, technical direction, engineering best practice, and quality/compliance. This also includes proper and effective architectural integration with other platforms, products, and services across the company. This leader will work closely with team memb [... source excerpt omitted ...] eep technical expertise in building and scaling data lake environments, coupled with strong people leadership and communication skills to drive alignment across the organization. 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 d
Employer postings · Data from · Sources