Junior AI/NLP/Machine Learning Engineer 2

WEX • Bay Area, CA

85d • $87,000 - $115,000

About The Position

We are a fast-growing AI Engineering team, driving WEX's strategic vision to integrate artificial intelligence into the core of our product and business. Our team thrives on collaboration, working directly with stakeholders and domain experts to develop innovative AI platforms and solutions that enhance decision-making and create groundbreaking products that benefit our customers and help them to effectively grow their business. We foster a culture of continuous learning and believe that the best work stems from a shared purpose and an enjoyable, engaging environment.

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 commonly used LLM models and/or ML packages including deep learning.
• Strong data manipulation skills, knowledge of data processing techniques and experience with ML frameworks.
• Solid understanding of probability and statistics concepts (distributions, Bayesian inference, maximum likelihood estimation, etc.)
• Experience with software development and the SDLC
• Hands-on experience with CI/CD tools and IaC (e.g., Terraform).
• Strong coding skills and a deep understanding of software design principles.
• Experience with Cloud technology such as AWS
• Familiar with Kubernetes.

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.

Benefits

• Health insurance
• Dental insurance
• Vision insurance
• Retirement savings plan
• Paid time off
• Health savings account
• Flexible spending accounts
• Life insurance
• Disability insurance
• Tuition reimbursement
• Quarterly or annual bonus based on role and applicable plan

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Industry: Credit Intermediation and Related Activities
• Education Level: Bachelor's degree
• Number of Employees: 5,001-10,000 employees