Role Description & Responsibilities
Machine Learning
About Stripe
Stripe is a high-growth technology leader building scalable solutions for global users. Millions of companies and teams rely on their infrastructure and services to execute high-value workflows, analyze massive datasets, and scale modern operations.
CadetX Candidate Advantage: As an applicant for this Machine Learning role, you are evaluated on practical proficiency with real-world query execution, data pipelines, and clear stakeholder communication.
CadetX Role Brief & Company Specifications
Role Mission & Impact
Stripe is a financial infrastructure platform for businesses. Millions of companies—from the world’s largest enterprises to the most ambitious startups—use Stripe to accept payments, grow their revenue, and accelerate new business opportunities. Our mission is to increase the GDP of the internet, and we have a staggering amount of work ahead. That means you have an unprecedented opportunity to put the global economy within everyone’s reach while doing the most important work of your career.
Growth Platform builds the machine learning systems that help businesses discover and use the Stripe products that meet their needs. Our recommendations reach users across the Dashboard, email, onboarding, documentation, and AI agent interfaces. We combine an understanding of each business with models that decide which recommendation is useful, when to show it, and how to learn from the outcome.
Key Responsibilities & Day-to-Day Scope
- ✓ Design, train, evaluate, deploy, and maintain models for recommendation, ranking, and personalized action selection across Growth Platform surfaces.
- ✓ Improve contextual bandit and policy-learning approaches, including exploration, reward design, and how recommendations adapt to user context and feedback.
- ✓ Build agent-based recommendation capabilities that use business context to identify relevant products and integration options, with evaluations that test recommendation quality and usefulness.
- ✓ Develop reliable data and feature pipelines for training and inference. Improve data freshness, feature quality, and consistency between training and production.
- ✓ Build reusable tooling for model evaluation, retraining, and safe rollout so the team can test and ship improvements faster.
- ✓ Own the quality and operation of the team's ML components: write tested production code, monitor models and pipelines, investigate failures, and improve reliability, latency, and cost.
- ✓ Design and analyze online experiments with data science partners. Connect offline evaluation to product adoption and incremental impact, with guardrails for dismissals, unsubscribes, and user experience.
- ✓ Partner with product engineering to integrate models into recommendation delivery systems, and with ML infrastructure teams to use and improve Stripe's shared training, feature, and serving capabilities.
- ✓ Work with product, marketing, and sales partners to identify problems that shared ML capabilities can solve, and make practical choices about where modeling adds value.
- ✓ 3+ years of industry experience in machine learning engineering, software engineering, or applied data science, with hands-on experience building and shipping ML models in production.
- ✓ Strong programming skills in Python and experience writing maintainable, tested production code.
Official Qualifications & Technical Requirements
- • Practical experience designing, training, and evaluating ML models using frameworks such as PyTorch, TensorFlow, XGBoost, or scikit-learn.
- • Experience building data or feature pipelines, proficiency in SQL, and familiarity with distributed data processing tools such as Spark or PySpark.
- • A strong understanding of statistics, model evaluation, and experimentation, including the ability to recognize data leakage and distinguish offline model improvements from business impact.
- • Experience deploying, monitoring, and debugging production ML systems, and evaluating tradeoffs among model quality, reliability, latency, and cost.
- • Ability to turn an open-ended business problem into a technical approach and collaborate effectively with engineering, data science, product, and business partners.
- • Experience with recommendation systems, ranking, personalization, or marketplace and advertising optimization.
- • Experience with contextual bandits, policy learning, causal inference, or off-policy evaluation.
- • Experience building and evaluating LLM applications, including structured extraction, embeddings, or recommendations grounded in user and business context.
- • Experience building reusable ML capabilities used by multiple products or teams, including training automation, feature systems, or model monitoring.
- • Experience with product growth, lifecycle messaging, or systems that balance short-term engagement with longer-term user outcomes.
Preferred Skills & Technology Stack
- + Familiarity with MLOps pipelines (Docker, Kubernetes, Kubeflow, Triton Inference Server).
- + Knowledge of distributed training techniques and GPU acceleration (CUDA).
- + Relevant coursework, research publication, or open-source contribution in ML/DL.
Tech Stack & Tools in this role
Expected technical competencies for Day 1 productivity:
★ Life & Culture at Stripe
Verified EmployerStripe champions rigorous analytical discipline, continuous learning, and low-ego teamwork. Team members enjoy flexible hybrid allowances, learning stipends, and structured mentorship to fast-track career mobility.