Staff Machine Learning Engineer

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

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

Experience you'll bring

• Minimum 7 years of application development experience.

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