AI Research Intern

The role

You will research and prototype transformer-based and graph-based foundation models using customer, transaction, and merchant data. You will work with product, engineering, and applied science teams to build working prototypes.

What you will do

• Research, design, and train foundation models on OnePay's data using transformer and/or graph-based architectures
• Run experiments and evaluate results rigorously
• Turn promising findings into a working prototype
• Partner with Product, Engineering and Analytics teams

Required qualifications

• Final-Year Graduate Student in the final year of a master’s or PhD program
• Expected to graduate in Fall 2027 or Spring 2028
• Majoring in Engineering or a related field
• Strong grasp of neural networks including backpropagation, attention, embeddings, optimization, and training dynamics
• Experience implementing, training, or fine-tuning models through research, coursework, or personal projects
• Comfortable working in PyTorch or an equivalent framework
• At least one previous internship in research, machine learning, or software engineering, or substantial personal or academic projects
• Available to work in the New York City office at least 3-4 days a week

Nice to have

• Familiarity with transformer-based foundation model architectures
• Exposure to graph-based learning or graph foundation models
• Coursework or projects involving Large Language Models or Generative AI
• Research output such as publications, workshop papers, or open-source contributions

Benefits

• Early access to a high potential, high growth fintech
• Hands on experience building product ready to ship
• Beautiful NYC SoHo Office
• Professional development and team social outings

Before you apply

• OnePay will not sponsor applicants who need employment sponsored visas
• No offer decisions will be made until the application window closes
• All candidates are evaluated together before final decisions go out

About the company

OnePay is a consumer fintech platform that brings together banking, high-yield savings, credit cards, point-of-sale lending, investing, and crypto. It is backed by Walmart and Ribbit Capital.