Mid-Level AI/ML/NLP Engineer

About the Team:

WEX Inc. is an innovative global commerce platform and payments technology company looking to forge the way in a rapidly changing environment, to simplify the business of doing business for customers, freeing them to spend more time, with less worry, on the things they love and care about. We are journeying to build a consistent world-class user experience across our products and services and leverage customer-focused innovations across all our strategic initiatives, including Big Data, AI, and Risk. Our AI Infrastructure team is pivotal in enabling these advancements.

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.

This is a really exciting time to be in the AI Infrastructure team at WEX. Our team is responsible for building and maintaining the robust, scalable, and secure cloud infrastructure that powers our AI and machine learning initiatives. We work with cutting-edge technologies like AWS, Azure, Docker, and Kubernetes to create a dynamic environment that supports the development and deployment of AI models at scale. We have challenging problems with huge business impact potential for you to work on and grow. We also have a strong team with highly talented and skillful engineers and leaders to support, guide, and coach you.

We are looking for a highly motivated and high-potential Engineer to join our AI Infrastructure team to make significant contributions to our cloud-based AI solutions and grow your career. If you dream to be a strong engineer who can solve tough problems, generate big impacts, and grow fast, this is a great opportunity for you!

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.

• Collaborate with partners and stakeholders to understand the requirements of our AI development teams and key challenges.

• Design, build, and maintain cloud infrastructure on AWS and Azure to support AI/ML workloads.

• Design, test, code, and instrument AI/ML platforms/tools.

• Develop, evaluate, deploy, and monitor AI/ML models.

• Implement Infrastructure as Code (IaC) using tools like Terraform.

• Support live platforms/systems, promote proactive monitoring, rapid incident response, and continuous improvement.

• Implement and manage containerization technologies (Docker) and orchestration platforms (Kubernetes).

• Develop and maintain CI/CD pipelines for automating the deployment and management of AI infrastructure.

• Develop and maintain monitoring and alerting systems to ensure the health and performance of production AI infrastructure.

• Analyze system performance data to identify bottlenecks and opportunities for improvement.

• Measure, inspect, and drive decisions using data.

• Mentor and learn from your peers, foster continuous learning of new cloud technologies and best practices.

• Get good at our team’s processes and best practices, and apply them to given tasks with help from peers and your manager; ensuring you understand the underlying problems you're trying to solve with these tasks, and that your implementations effectively address these problems in a reliable and sustainable way.

• Partner with team members in development and problem-solving.

• Independently complete work and proactively seek reviews from Senior Engineers and Leaders on your work to ensure high quality.

• Perform technical discussions. Proactively review work from peers and learn.

• 2+ years of experience as an AI/ML Engineer or Data Scientist with strong software development skills.

• Bachelor's degree in Computer Science, Software Engineering, or related field.

• A Master's or PhD degree in Computer Science, Machine Learning, or a related field is a plus!

• Experience with software development and the SDLC.

• Proven track record of delivering end-to-end solutions using diverse technologies, and proficiency with at least one of the following 3GL strongly-typed languages as your core programming language: Java, C/C++/C#, Golang, or Python.

• Strong understanding of cloud platforms (AWS and Azure), including services relevant to AI/ML (e.g., EC2, S3, EKS, Azure ML, AKS).

• Hands-on experience with containerization (Docker) and container orchestration (Kubernetes).

• Experience with building and managing CI/CD pipelines for infrastructure and ML model deployment (using tools like Jenkins, GitLab CI/CD, GitHub Actions, etc.).

• Strong understanding of networking concepts (VPC, subnets, routing, firewalls) and experience configuring network infrastructure in the cloud.

• Experience with infrastructure monitoring and alerting tools (e.g., Prometheus, Grafana, CloudWatch, Azure Monitor).

• Strong scripting skills (Python, Bash) for automation and configuration management.

• Excellent problem-solving skills, with the ability to analyze complex systems and identify performance bottlenecks.

• Experience with using ML algorithms and platforms.

• Experience with using Large Language Models for generative AI projects.

• ​Experience in commonly used LLM's and/or ML packages, including deep learning.

• Understanding of data processing techniques.

• Strong data manipulation skills.

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

• Strong communication and collaboration skills, with the ability to work effectively in a team environment.

• Demonstrated ability to collaborate with stakeholders throughout the product delivery lifecycle.

• Evidence of innovative problem-solving skills.

• Experience in the financial industry is a plus.