Principal AI/ML Engineer

WEX Inc.about 1 year ago
San Francisco, California, United States
Remote
Full-time
Junior Level (1-3 years)

Job Description

Position Overview

We are seeking an experienced and innovative Principal AI/ML Engineer to lead and drive the development of our AI/ML technology platform. You will design and implement cutting‐edge infrastructure and services while guiding your team to deliver high‐quality technical solutions that address broad risk challenges across the organization.

Key Responsibilities

  • Lead and drive the development of technology and platform for the company's AI/ML engineering needs, ensuring functional richness, reliability, performance, and flexibility.
  • Design the architecture and lead the implementation of the AI/ML infrastructure, platform, and services.
  • Challenge the status quo and continuously drive for better technical solutions in the broad risk area of the organization.
  • Guide your team to leverage data and AI tools in design and development, providing strong thought leadership and technical mentorship.
  • Partner with senior engineers across the organization to understand business needs and jointly develop effective technical solutions.
  • Collaborate with information security teams to ensure data security and privacy through proper encryption and access controls.
  • Work with solutions architecture and product/engineering groups on integrating the AI/ML Platform with other systems for reliable execution.
  • Engage with external partners and vendors to evaluate complementary technologies and services.
  • Create, prioritize, communicate, and execute roadmaps, project plans, and commitments.
  • Report on development status, quality, operations, and system performance to management and partners.
  • Drive the technical vision and strategy for AI/ML infrastructure for both near-term and long-term business needs.
  • Provide technical guidance, training, and mentorship to engineering teams, and uphold high standards including DORA metrics.
  • Set clear, measurable quality goals and partner with security and SRE teams to ensure continuous improvement and compliance with SLOs.
  • Take strong financial responsibility for your team and hold them accountable for delivering quality results using defined OKRs.
  • Advise senior leadership on complex situations and lead teams through initiatives to achieve defined OKRs.
  • Provide strategic thought leadership on business initiatives using deep technical and industry expertise.

Required Qualifications

  • A technically deep and innovative Software Engineering leader with 15-20 years of design and development experience.
  • Strong software development skills in a chosen programming language with willingness to learn Java and Python.
  • Extensive experience in building machine learning models (both supervised and unsupervised) with familiarity in Reinforcement Learning.
  • Proven expertise with distributed computing frameworks such as Ray and Spark, Kubernetes, Terraform, and cloud architecture tools (AWS, Azure).
  • Hands-on experience in architecture design, development, deployment, and production operations within an agile environment, emphasizing automation, CI/CD, and quality best practices.
  • Deep understanding of Big Data, AI/ML platforms, and large-scale data processing.
  • Excellent analytical, problem-solving, and strong written and verbal communication skills.
  • Ability to mentor and inspire teams while fostering a culture of transparency and continuous improvement.
  • Bachelor degree in Computer Science or a related field; advanced degree is a plus.

Preferred Qualifications

  • Experience in Reinforcement Learning and using Large Language Models (LLMs).
  • Experience with Sagemaker and similar AI/ML platforms.
  • Advanced degree in Computer Science or a related field.

Leadership & Personal Characteristics

  • Demonstrates a collaborative, engaging, mission-driven style with strong listening skills, humility, empathy, transparency, and cultural sensitivity.
  • Champions a culture of high standards, progress, and change while upholding uncompromising ethics and financial responsibility.
  • Excels in a fast-paced entrepreneurial environment with a hands-on and practical approach.
  • Communicates effectively in both small and large group settings, articulating a compelling vision that excites teams.
  • Exhibits intellectual curiosity, learning agility, and strong stakeholder management skills to drive effective change.

Benefits & Perks

  • Compensation: Base pay range: $203,000.00 - $270,000.00
  • Benefits: Comprehensive, market-competitive benefits package including health, dental, and vision insurances, retirement savings plan, paid time off, health savings account, flexible spending accounts, life and disability insurance, tuition reimbursement, and more.

Required Skills

Terraform
Machine Learning
Data Science
Mentorship
Kubernetes
Distributed Computing (Ray, Spark)
Big Data Platforms
Leadership
Cloud Architecture (AWS, Azure)
AI/ML Engineering
CI/CD Automation
Software Architecture
Agile Development