WEX Inc is hiring an AI/ML/NLP Engineer Intern as a PhD Research Intern. You will apply deep technical expertise to solve complex, real-world problems, owning a project from research and design through to a functional prototype. This role is for a highly motivated researcher passionate about applying AI, Machine Learning, and software engineering to create tangible business value.
What You'll Do
- Collaborate with senior engineers, product managers, and domain experts to understand our customers' needs and challenges.
- Apply your research skills to design, code, and test novel AI/ML solutions.
- Develop, evaluate, and iterate on AI/ML models (including LLMs) to solve a core business problem.
- Analyze existing platforms to identify bottlenecks and propose opportunities for algorithmic or systems-level improvements.
- Gain hands-on experience with our production systems, including CI/CD pipelines (GitHub Actions) and Infrastructure as Code (Terraform).
- Present your findings and project outcomes to the team and key stakeholders at the end of your internship.
What We're Looking For
- Currently enrolled in a PhD program in Computer Science, Artificial Intelligence, Machine Learning, Statistics, or a related quantitative field.
- Demonstrated research experience in AI, Machine Learning, Deep Learning, or Generative AI (e.g., publications, conference papers, or significant project work).
- Strong programming skills in at least one of our core languages: Python, Java, C#, or GoLang.
- Deep understanding of modern ML algorithms, platforms, and packages (including deep learning and LLMs).
- Solid grasp of probability and statistics concepts (e.g., distributions, Bayesian inference, maximum likelihood estimation).
- A collaborative mindset and strong communication skills, with an ability to translate complex research into practical applications.
- Evidence of innovative problem-solving and a drive to learn and excel.
Nice to Have
- Experience with cloud platforms (e.g., AWS).
- Familiarity with software development best practices (SDLC), CI/CD tools, and IaC (e.g., Terraform).
- Familiarity with container orchestration (e.g., Kubernetes).
- Experience or interest in the financial technology (FinTech) industry.
Technical Stack
- Languages: Python, Java, C#, GoLang
- Cloud & Infrastructure: AWS, GitHub Actions, Terraform, Kubernetes
Team & Environment
You will be paired with a dedicated mentor and work within the fast-growing AI Engineering team, collaborating with senior engineers, product managers, and domain experts.
Benefits & Compensation
- Compensation: $35.00 - $43.00 per hour
- Health, dental and vision insurances
- Retirement savings plan
- Paid time off
- Health savings account
- Flexible spending accounts
- Life insurance
- Disability insurance
- Tuition reimbursement
Company Culture
WEX Inc thrives on collaboration and working directly with stakeholders and domain experts. We foster a culture of continuous learning and believe the best work stems from a shared purpose and an enjoyable, engaging environment.
