Mid AI/ML Engineer

WEX - San Francisco, CA

Full-time - Mid Level

Remote - San Francisco, CA

Computing Infrastructure Providers, Data Processing, Web Hosting, and Related Services

About the position

The Mid AI/ML Engineer at WEX, Inc. is a pivotal role within a fast-growing AI Engineering team focused on integrating artificial intelligence into the company's products and business strategies. This remote position requires collaboration with stakeholders to develop innovative AI platforms and solutions that enhance decision-making and drive business growth. The ideal candidate is passionate about AI and Machine Learning, possesses strong software engineering skills, and is motivated to solve complex problems using data-driven approaches.

Responsibilities

• Collaborate with stakeholders to learn about our customers' needs and challenges.
• Measure, inspect, and drive decisions using data.
• Design, test, code, and instrument AI/ML platforms/tools.
• Develop, evaluate, deploy, and monitor AI/ML models.
• Develop and maintain CI/CD pipelines using tools such as GitHub Actions.
• Implement Infrastructure as Code (IaC) using tools like Terraform.
• Support live platforms/systems, promote proactive monitoring, rapid incident response, and continuous improvement.
• Analyze existing platforms/systems and processes to identify bottlenecks and opportunities for improvements.

Requirements

• 2+ years of experience as an AI/ML engineer or Data Scientist with strong software development skills or PhD in AI/ML
• Bachelor's degree in Computer Science, Software Engineering, or related field; A Master or PhD in the area of AI/ML is a big plus
• Proven track record of delivering end-to-end solutions using diverse technologies, with Java, C#, GoLang, or Python as your core programming language
• Demonstrated ability to collaborate with stakeholders throughout the product delivery lifecycle
• Evidence of innovative problem-solving skills
• Experience with using Large Language Models for generative AI projects
• Experience with using ML algorithms and platforms
• Understanding of data processing techniques
• Experience in the financial industry a plus

Nice-to-haves

• Experience with Cloud technology such as AWS
• Familiar with Kubernetes

Benefits

• life insurance
• paid time off
• tuition reimbursement
• health insurance
• dental insurance
• vision insurance
• disability insurance
• retirement savings plan
• health savings account
• flexible spending accounts