Note: The job is a remote job and is open to candidates in USA. WEX is seeking a PhD Research Intern to join its AI Engineering team and help integrate artificial intelligence into its products and business. The intern will conduct research, design and develop AI/ML solutions including LLMs, improve existing platforms, gain experience with production systems, and present project outcomes to stakeholders.
Responsibilities
- 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
Skills
- 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
- 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
Benefits
- Remote position; candidate must reside within 30 miles of Portland, ME; Boston, MA; Chicago, IL; Dallas, TX; San Francisco Bay Area, CA; or Seattle, WA.
- Dedicated mentor and a high-impact project to own.
- Hands-on experience with production systems, including CI/CD pipelines and Infrastructure as Code.
- Health, dental, and vision insurance.
- Retirement savings plan.
- Paid time off.
- Health savings account.
- Flexible spending accounts.
- Life insurance.
- Disability insurance.
- Tuition reimbursement.
Company Overview
Company H1B Sponsorship