Staff Machine Learning Engineer

WEX, Inc.; San Francisco (CA); On-site; USD 147,000 - 195,000; Full time

Job summary

Join a forward-thinking company as a Machine Learning Engineer, where you'll work on cutting-edge AI technologies and collaborate with talented teams. This role offers the chance to design and implement machine learning algorithms, develop RESTful APIs, and contribute to a cloud-first environment. If you're passionate about innovation and thrive in a fast-paced setting, this opportunity is perfect for you. Enjoy a competitive salary and comprehensive benefits while making a significant impact in the financial industry.

Benefits

Health Insurance

Dental Insurance

Vision Insurance

Retirement Savings Plan

Paid Time Off

Tuition Reimbursement

Life Insurance

Disability Insurance

Flexible Spending Accounts

Health Savings Account

Qualifications

• 7+ years of application development experience.
• 4+ years of experience building and deploying machine learning models.
• Familiarity with cloud technologies and IAC frameworks.

Responsibilities

• Design and implement machine learning algorithms and pipelines.
• Collaborate with cross-functional teams to integrate AI components.
• Manage version control and implement CI/CD pipelines.

Skills

Python

Machine Learning

DevOps

RESTful APIs

Version Control (GitHub)

Problem Solving

Education

Bachelor's Degree in Computer Science

Tools

AWS

Azure

GCP

Terraform

Pandas

Numpy

PyTorch

TensorFlow

Job description

(*) The candidate must reside within 30 miles of one of the following locations: San Francisco Bay Area, CA; Portland, ME; Boston, MA; Chicago, Il.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 thirst for working with new tools and technologies, then this is the place to be!

How you'll make an impact

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

The base pay range represents the anticipated low and high end of the pay range for this position. Actual pay rates will vary and will be based on various factors, such as your qualifications, skills, competencies, and proficiency for the role. Base pay is one component of WEX's total compensation package. Most sales positions are eligible for commission under the terms of an applicable plan. Non-sales roles are typically eligible for a quarterly or annual bonus based on their role and applicable plan. WEX's comprehensive and market competitive benefits are designed to support your personal and professional well-being. Benefits include 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. For more information, check out the "About Us" section.Pay Range: $147,000.00 - $195,000.00