Posted in

Staff AI Infra Engineer

Staff AI Infra Engineer

CompanyWex
LocationSan Francisco, CA, USA, Chicago, IL, USA, Portland, ME, USA
Salary$150000 – $200000
TypeFull-Time
DegreesBachelor’s
Experience LevelSenior, Expert or higher

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.

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.

Preferred Qualifications

  • Experience in the financial industry is a plus.