AI Research, Intern

OnePay

2d•$32 - $32•Onsite

About The Position

As an AI Research Intern, your mandate is to research and prototype foundation models, both transformer-based and graph-based, applied to OnePay's customer, transaction, and merchant data. You'll work on real modeling problems and build a working prototype that demonstrates the approach. This role will impact OnePay's vision by providing a remarkable customer experience, driving positive business outcomes, and improving operational efficiency. You will work closely with the product, engineering and applied science team(s) working across all business units. This role reports to the Head of AI/ML. This role is responsible for: Research, design, and train foundation models on OnePay's data, working with transformer and/or graph-based architectures. Run experiments, evaluate results rigorously, and turn promising findings into a working prototype. Opportunity to work in an extremely fast paced growing company Partner with Product, Engineering and Analytics teams to ensure we are providing a remarkable customer experience. Driving significant impact on the company’s bottom line.

Requirements

Final-Year Graduate Student: You’re 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.
Deep Learning Fundamentals: You have a strong grasp of how neural networks work under the hood, including backpropagation, attention, embeddings, optimization, and training dynamics. You've implemented, trained, or fine-tuned models yourself through research, coursework, or personal projects, and you're comfortable working in PyTorch or an equivalent framework.
Prior Experience: You've completed at least one previous internship in research, machine learning, or software engineering, or have built substantial personal or academic projects you're excited to talk through.
Availability: This is a full-time, 8-week program running June 21 - August 13, 2027. You are available to work in our New York City office at least 3-4 days a week throughout.
Proactive & Collaborative Attitude: You have an act-like-an-owner mentality, thrive in a fast-moving environment, and enjoy working with others to achieve collective goals.
Critical Thinking & Problem-Solving Skills: You enjoy tackling complex challenges, analyzing problems, and developing solutions. Clear communication of ideas is one of your strengths.
Organized & Detail-Oriented: You’re a keen listener with strong organizational skills, capable of managing multiple tasks efficiently.
Excited to Learn & Grow: You are passionate about being part of a rapidly growing start-up and look forward to expanding your skills in a real-world setting.

Nice To Haves

Familiarity with transformer-based foundation model architectures, including pre-training, fine-tuning, inference, and the tradeoffs between approaches.
Exposure to graph-based learning or graph foundation models, such as GNNs, message passing, or representation learning on graph-structured data.
Coursework or projects involving Large Language Models or Generative AI.
Research output in any of the above: publications, workshop papers, or open-source contributions.

Responsibilities

Research, design, and train foundation models on OnePay's data, working with transformer and/or graph-based architectures.
Run experiments, evaluate results rigorously, and turn promising findings into a working prototype.
Partner with Product, Engineering and Analytics teams to ensure we are providing a remarkable customer experience.
Driving significant impact on the company’s bottom line.