Apply on company website United States USD 72,800 – 89,440 / year

WEX Inc is hiring an AI/ML/NLP Engineer Intern

As an AI/ML/NLP Engineer Intern, you'll take part in a hands-on program centered on creating intelligent systems using state-of-the-art machine learning and natural language processing methods. You'll partner with seasoned engineers and subject matter experts to identify key problems and develop data-driven solutions that have measurable impact.

What You'll Do

  • Work closely with technical leads and product teams to uncover customer pain points and shape model-driven responses.
  • Design, implement, and validate machine learning approaches, including large language models, through rigorous experimentation.
  • Evaluate current platforms for inefficiencies and recommend algorithmic or architectural optimizations.
  • Engage with production workflows, including automated testing and deployment systems like GitHub Actions and infrastructure managed through Terraform.
  • Communicate technical results clearly to both technical and non-technical audiences at the close of your project cycle.

What We're Looking For

  • Enrollment in a PhD program in Computer Science, AI, Machine Learning, Statistics, or a closely related quantitative discipline.
  • Proven research background in areas such as deep learning, generative AI, or statistical modeling, demonstrated through publications or substantial projects.
  • Proficiency in at least one major programming language—Python, Java, C#, or GoLang—with a preference for Python in ML contexts.
  • Strong foundation in core machine learning frameworks and methodologies, including experience with deep neural networks and transformer-based architectures.
  • Working knowledge of statistical theory, including probability distributions, Bayesian reasoning, and estimation techniques.
  • Ability to collaborate effectively and articulate complex technical concepts in accessible ways.
  • A track record of creative problem solving and self-directed learning.

Preferred Background

  • Exposure to cloud environments, particularly AWS.
  • Understanding of software development lifecycles, version control, CI/CD pipelines, and Infrastructure as Code tools like Terraform.
  • Basic familiarity with containerized systems and orchestration platforms such as Kubernetes.
  • Interest or prior experience in financial technology domains.

Tools & Technologies

Python, Java, C#, GoLang, GitHub Actions, Terraform, AWS, Kubernetes, Machine Learning, Deep Learning, Large Language Models (LLMs), CI/CD, Infrastructure as Code (IaC)

Benefits

  • Health, dental, and vision insurance
  • Retirement savings plan
  • Health savings and flexible spending accounts
  • Life and disability insurance
  • Paid time off
  • Tuition reimbursement

Compensation

Hourly rate between $35.00 and $43.00. Some roles may be eligible for performance-based bonuses depending on the position and plan.

Required Skills
PythonJavaC#GoMachine LearningDeep LearningAWSKubernetesTerraformGitHub Actions
Job Details
Location United States
Salary USD 72,800 – 89,440 / year
Department Research and Development (R&D)
Category Data & ML
Posted 5 months ago
Application On company website
About company
WEX Inc logo
WEX simplifies the business of running a business. Their WEX Benefits solutions reduce complexity and help manage costs of benefits administration for clients and partners.
All jobs at WEX Inc Visit website