Senior Machine Learning Engineer

About the Team/Role

We are the AI Engineering team at WEX. We enable our product development teams to move quickly and deploy AI functionality into their applications in a scalable manner. We are cloud-first and aim to automate everything, including infrastructure, testing, and deployment. If you’re interested in ChatGPT, Large Language Models, MLOps, DevOps, and have a thirst for working with new tools and technologies, then this is the place to be!

• A highly motivated engineer who loves working on small, highly performing teams.
• Collaborative, a solid communicator, and works well with your team and stakeholders.
• A professional who can advocate strongly for positions and still be 100% behind team decisions even if they don’t go your way.
• Comfortable balancing the need to move fast with the realities of working in highly regulated spaces like payments and healthcare.
• Design, implement, train, and maintain machine learning algorithms and ML system pipelines.
• Work closely with our AI platform team and line of business engineering teams to integrate AI components and models into WEX systems.
• Collaborate with the team to manage version control using GitHub and implement robust CI/CD pipelines for continuous integration and delivery.
• Design and implement RESTful APIs to facilitate communication between various components and ensure smooth operation of our applications.
• Collaborate effectively with cross-functional teams, participate in code reviews, and communicate solutions to both technical and non-technical stakeholders.

• Minimum 5 years of application development experience using Python.
• Minimum 3 years of experience building, training, and deploying machine learning models.
• Strong ability to use key Python libraries such as Pandas, Numpy, and deep learning libraries like PyTorch or TensorFlow.
• Bachelor's Degree in Computer Science, Engineering, or a related field.
• Familiarity with cloud technologies, such as AWS, Azure, or GCP, and how to provision those services through IAC frameworks such as Terraform.
• Experience applying DevOps principles to software development processes.
• Proficiency in version control systems, particularly GitHub.
• Excellent problem-solving skills and a proactive approach to addressing challenges.
• Experience in the financial industry is a plus.