Staff AI Infrastructure Engineer (Software Engineering Focus)

About the Role

Key Responsibilities

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

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.

Nice to Have

• Experience in the financial industry is a plus.

Qualifications

• Bachelor's Degree in Computer Science, Engineering, or related field.

Benefits & Perks

• Health, dental and vision insurances
• Retirement savings plan
• Paid time off
• Health savings account
• Flexible spending accounts
• Life insurance
• Disability insurance
• Tuition reimbursement