AI Infrastructure Engineer

About the Team/Role

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.

How You’ll Make an Impact

You are a highly motivated and results-oriented individual with a passion for AI, Machine Learning, and software engineering. You possess a strong drive to learn and excel in solving complex problems using data-driven tools and algorithms. You understand the importance of effective communication and collaboration in achieving successful outcomes.

How you’ll make an impact

• 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.

Experience You’ll Bring

• A Master's or PhD in an AI/ML-related field is strongly preferred.

• Alternatively, a Bachelor's degree in Computer Science, Software Engineering, or a related field combined with a minimum of 2+ years of professional experience as an AI/ML engineer or a Data Scientist with strong software development skills.

• 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 is a plus.

Technical Skills

• Proficiency in at least one of the languages mentioned above.

• Experience in commonly used LLM models and/or ML packages including deep learning.

• Strong data manipulation skills.

• Solid understanding of probability and statistics concepts (distributions, Bayesian inference, maximum likelihood estimation, etc.).

• Experience with software development and the SDLC.

• Experience with Cloud technology such as AWS.

• Familiarity with Kubernetes.