Machine Learning Engineer, Foundation Model jobs in United States
cer-icon
Apply on Employer Site
company-logo

Stripe · 4 days ago

Machine Learning Engineer, Foundation Model

Stripe is a company focused on accelerating global economic and technological development through financial infrastructure. The Machine Learning Engineer on the Foundation Model team will tackle challenging technical problems and contribute to the development of foundational models that enhance Stripe's payments and risk solutions.

FinanceFinTechMobile PaymentsSaaS
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Develop foundation models for payments, merchants, and consumers that span Stripe product areas
Fine-tune and optimize LLMs using SOTA techniques, including multimodal alignment, knowledge distillation, quantization, and reasoning to build highly specialized models
Drive technical excellence through hands-on contributions to the design and development of state-of-the-art AI/ML systems, conducting architecture reviews, and maintaining high code quality
Partner with engineering and product leaders across Stripe to identify and prioritize foundational investments such as foundation models, assistants, and agents that unlock new capabilities for product teams
Contribute to Stripe's technical strategy by representing AI/ML engineering perspectives in company-wide technical decisions and roadmap planning
Mentor ML engineers across Stripe on ML experimentation, helping teams navigate complex technical trade-offs, and adopt platform capabilities effectively

Qualification

Machine LearningPythonFoundation ModelsDeep Neural NetworksTransformersSOTA TechniquesExperimental MindsetResearch SkillsProject ManagementMentoringProblem Solving

Required

5+ years of experience building and shipping ML models that power AI/ML product features, with a strong emphasis on modern technologies such as DNNs, Transformers, and Foundation Models
Strong programming skills in Python with demonstrated ability to write production-quality code
A strong experimental mindset, with a proven track record of identifying, prototyping, and validating SOTA machine learning methods to enhance model performance
Proven ability to shepherd large, complex ML projects and drive transformational change in an organization
Deep passion for solving really interesting problems and for building the latest technologies rather than relying on outdated methods

Preferred

A PhD or Master's degree with a research-oriented background, with the ability to dive into research papers and stay current with academic publications
Experience with a large-scale, data-rich product in a domain such as payments, commerce, search, or social media
Knowledge of the challenges and opportunities in applying ML to fraud prevention, merchant intelligence, or financial services
Published research or open source contributions in AI/ML or related fields

Benefits

Equity
Company bonus or sales commissions/bonuses
401(k) plan
Medical, dental, and vision benefits
Wellness stipends

Company

Stripe is an API technology company that provides online payment processing and commerce solutions for Internet businesses.

H1B Sponsorship

Stripe 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 (272)
2024 (283)
2023 (141)
2022 (379)
2021 (276)
2020 (111)

Funding

Current Stage
Late Stage
Total Funding
$9.44B
Key Investors
Tiger Global ManagementThrive CapitalFounders Fund
2025-02-27Secondary Market
2024-04-12Secondary Market· $694.16M
2024-03-01Series Unknown

Leadership Team

leader-logo
Steffan Tomlinson
CFO
linkedin
leader-logo
Aitor Guevara
Head Of Engineering
linkedin
Company data provided by crunchbase