Staff Machine Learning Engineer

$147,000 - $195,000/Yr

Wex Corporate Payments - Portland, ME

Full-time - Mid Level

Portland, ME

Funds, Trusts, and Other Financial Vehicles

About the position

We are the AI Engineering team at WEX, dedicated to enabling our product development teams to rapidly deploy AI functionality into their applications in a scalable manner. Our approach is cloud-first, and we strive to automate every aspect of our operations, including infrastructure, testing, and deployment. If you have a keen interest in cutting-edge technologies such as ChatGPT, Large Language Models, MLOps, and DevOps, this is the ideal environment for you to thrive and innovate. In this role, you will be a highly motivated engineer who enjoys working within small, high-performing teams. You will be expected to collaborate effectively, communicate clearly, and advocate for your ideas while supporting team decisions. Balancing the need for speed with the realities of working in regulated environments like payments and healthcare will be crucial. Your responsibilities will include designing, implementing, training, and maintaining machine learning algorithms and ML system pipelines. You will work closely with our AI platform team and business engineering teams to integrate AI components and models into WEX systems. Additionally, you will manage version control using GitHub and implement robust CI/CD pipelines for continuous integration and delivery. Designing and implementing RESTful APIs to facilitate communication between various components will also be part of your duties. You will collaborate with cross-functional teams, participate in code reviews, and communicate solutions to both technical and non-technical stakeholders.

Responsibilities

• Design, implement, train and maintain machine learning algorithms and ML system pipelines.
• Work closely with the 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.

Requirements

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

Benefits

• Health insurance
• Dental insurance
• Vision insurance
• Retirement savings plan
• Paid time off
• Health savings account
• Flexible spending accounts
• Life insurance
• Disability insurance
• Tuition reimbursement