[Remote] AI Principal Engineer

Note: The job is a remote job and is open to candidates in USA. WEX is a fast-growing commerce platform and payment technology company, seeking a Principal AI/ML Engineer to lead the design and implementation of their AI/ML engineering platform. The role involves driving technology development, architecture, and mentoring while collaborating with various teams to meet business needs and enhance data infrastructure.ResponsibilitiesLead 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 platformHelp design the architecture and lead the implementation of the AI/ML infrastructure, platform and servicesChallenge 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 WexGuide 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 areaClosely 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 partnershipPartner closely with information security teams to ensure the security and privacy of data, implementing encryption and access controls as requiredPartner 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 executionEngage with external partners and vendors to evaluate technologies and services that complement the company’s AI/ML strategyCreate, prioritize, communicate, manage, and execute roadmaps, project plans, and commitmentsReport on the status of development, quality, operations, and system performance to management and partnersDrive 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 applicationsSet 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 sameSet clear, measurable quality goals for your group in a data-driven wayPartner with the security org and SRE org to ensure continuous improvement and meeting of security requirements, SLOs, and error budgetTake strong financial responsibility for your teamHold yourself and your team accountable for delivering quality results using defined OKRsAdvise 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 organizationProvide strategic thought leadership on business initiatives using your deep technical and business industry expertiseSkillsA technically deep and innovative Software Engineer leader able to act and deliver to the business needsHave 15-20 years of software design and development experience at a large scaleStrong software development skills in his or her chosen programming language, but ready and willing to learn and program in Java and PythonGood experience in building machine learning supervised and unsupervised models. Experience in Reinforcement Learning is a plusExperience in using LLM models a plusBroad understanding of distributed computing frameworks like Ray and SparkDemonstrable experience in Kubernetes and TerraformExperience with Sagemaker a plusBroad understanding of cloud architecture tools and services, such as S3, EMR, Kubernetes, Lambda functions. Experience in AWS and Azure is highly desirableDeep 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 monitoringRich experience and deep expertise in the areas of Big Data and large scale data platforms, especially in AI and MLHighly experienced and deep understanding of CI/CD automationRich experience in combined engineering practice and agile development. Experienced in leading and guiding teams for effective development using this approachExcellent analytical skills; Capability in identifying and resolving complex problemsA mentor to individual contributors of different seniority levels and expertiseAbility to encourage and foster a culture of visibility and transparency across teamsStrong written and verbal communication skillsStrong philosophy of continuous improvementBachelor degree in Computer Science or a related fieldExperience in Reinforcement Learning is a plusExperience with Sagemaker a plusAn understanding of and experience in payment organizations is a plusA higher degree in a related field is a plusBenefitsMost 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.Company OverviewWex is a financial technology service provider for fleet, travel and healthcare industries. It was founded in 1983, and is headquartered in Portland, Maine, USA, with a workforce of 5001-10000 employees. Its website is http://www.wexinc.com.Company H1B SponsorshipWEX has a track record of offering H1B sponsorships, with 4 in 2026, 22 in 2025, 10 in 2024, 12 in 2023, 20 in 2022, 14 in 2021, 18 in 2020. Please note that this does not guarantee sponsorship for this specific role.