Logo
WEX, Inc.

Principal AI/ML Engineer

WEX, Inc., San Francisco

Save Job

Responsibilities

  • Lead the development of technology and platform for the company's AI/ML engineering needs, ensuring functionality, reliability, performance, and flexibility.
  • Design architecture and lead implementation of AI/ML infrastructure, platform, and services.
  • Challenge existing solutions and drive for improved technical solutions across the broad risk area of Wex.
  • Guide teams to leverage data and AI tools, demonstrating thought leadership and influencing peers.
  • Partner with other senior engineers to understand business needs and develop effective technical solutions.
  • Collaborate with security teams to ensure data security and privacy, including encryption and access controls.
  • Work with solutions architecture and LOB teams on AI/ML platform integrations, ensuring reliable execution.
  • Engage with external partners to evaluate complementary technologies and services.
  • Create and manage roadmaps, project plans, and commitments, reporting progress to stakeholders.
  • Drive the technical vision for AI/ML infrastructure aligning with business needs, in partnership with Data Science and ML teams.
  • Set high standards and provide technical guidance, mentorship, and training to engineering teams, fostering a culture of continuous learning and innovation.
  • Establish clear, measurable quality goals for the team using data-driven approaches.
  • Partner with security and SRE teams to meet security requirements, SLOs, and error budgets.
  • Take financial responsibility for the team and ensure accountability through OKRs.
  • Advise senior leadership on complex issues and lead initiatives to achieve organizational goals.
  • Provide strategic technical leadership on business initiatives, leveraging deep industry and technical expertise.

Qualifications & Experience

  • Deep technical leadership in software engineering, capable of delivering aligned with business needs.
  • 15-20 years of experience in large-scale software design and development.
  • Strong skills in programming languages, with willingness to learn Java and Python.
  • Experience in building supervised and unsupervised machine learning models; reinforcement learning experience is a plus.
  • Knowledge of LLM models is advantageous.
  • Understanding of distributed computing frameworks like Ray and Spark.
  • Experience with Kubernetes and Terraform; familiarity with AWS services such as Sagemaker, S3, EMR, Lambda, and Azure is desirable.
  • Hands-on experience in architecture, development, deployment, and operations; familiar with agile practices and CI/CD automation.
  • Deep expertise in Big Data, AI, and ML platforms.
  • Strong analytical, problem-solving, and communication skills.
  • Understanding of payment organizations is a plus.
  • Mentorship skills across different seniority levels.
  • Ability to foster transparency and a culture of continuous improvement.
  • Bachelor's degree in Computer Science or related field; higher degrees are a plus.

Leadership & Personal Characteristics

  • Collaborative, engaging, and mission-driven leadership style with strong listening skills.
  • Champion high standards, progress, and change, upholding ethics and integrity.
  • Operate effectively in fast-paced, entrepreneurial environments with a hands-on approach.
  • Excellent communicator with the ability to articulate vision and motivate teams.
  • High intellectual curiosity, learning agility, and change management skills.
  • Skilled in stakeholder management and navigating complex environments.
#J-18808-Ljbffr