[Remote] Mid-Level AI/ML/NLP Engineer

Note: The job is a remote job and is open to candidates in USA. WEX Inc. is an innovative global commerce platform and payments technology company looking to simplify the business of doing business for customers. The role involves collaborating with partners to design and maintain cloud infrastructure for AI/ML workloads, developing and monitoring AI/ML models, and supporting continuous improvement in a fast-growing AI Engineering team. Responsibilities • 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. Skills • 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. • 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. • A Master's or PhD degree in Computer Science, Machine Learning, or a related field is a plus! • Experience in the financial industry is a plus. Benefits • Health, dental and vision insurances • Retirement savings plan • Paid time off • Health savings account • Flexible spending accounts • Life insurance • Disability insurance • Tuition reimbursement • And more Company Overview • Wex is a financial technology service provider for fleet, travel and healthcare industries. It was founded in 1983, and is headquartered in Portland, Maine, USA, with a workforce of 5001-10000 employees. Its website is Company H1B Sponsorship • WEX has a track record of offering H1B sponsorships, with 17 in 2025, 10 in 2024, 12 in 2023, 20 in 2022, 14 in 2021, 18 in 2020. Please note that this does not guarantee sponsorship for this specific role. Apply tot his job