PhD Intern - AI/ML/NLP Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

WEX · 4 days ago

PhD Intern - AI/ML/NLP Engineer

WEX is a fast-growing AI Engineering team focused on integrating artificial intelligence into their products and business. As a PhD Research Intern, you will apply your technical expertise to solve complex problems and develop innovative AI solutions while collaborating with senior engineers and domain experts.

Financial ServicesFinTechPaymentsSoftware
check
H1B Sponsor Likelynote

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

Qualification

AIMachine LearningDeep LearningPythonCI/CDTerraformStatisticsFinTechCloud platformsKubernetesCollaborationCommunicationProblem-solving

Required

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

Preferred

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

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

Company

Wex is a financial technology service provider for fleet, travel and healthcare industries.

H1B Sponsorship

WEX has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (22)
2024 (10)
2023 (12)
2022 (20)
2021 (14)
2020 (18)

Funding

Current Stage
Public Company
Total Funding
$900M
Key Investors
Warburg Pincus
2025-02-25Post Ipo Debt· $500M
2020-06-30Post Ipo Equity· $90M
2020-06-30Post Ipo Debt· $310M

Leadership Team

leader-logo
Melissa Smith
Chair and Chief Executive Officer
linkedin
leader-logo
Sachin Dhawan
Chief Technology Officer
linkedin
Company data provided by crunchbase