Staff ML Engineer, Open Banking Data at Scale

Stripe; San Francisco (CA); Hybrid; USD 254,000 - 380,000; Full time

Benefits offered by this job

Equity

401(k) plan

Medical, dental, vision benefits

Wellness stipends

Job summary

Stripe's Financial Connections team is hiring a senior ML engineer to design, train, and deploy production ML models at scale across thousands of financial institutions. You will own end-to-end ML pipelines, focusing on data quality, risk scoring, and enrichment for merchants and consumers, while collaborating with product, data science, and engineering partners.

You will mentor engineers and help grow a strong ML culture.

Qualifications

• 10+ years of industry experience building and shipping ML systems in production.
• Proficient with ML libraries and frameworks such as PyTorch, TensorFlow, XGBoost, as well as Spark.
• Hands-on experience in designing, training, and evaluating machine learning models.
• Hands-on experience in productionizing and deploying models at scale.
• Hands-on experience in orchestrating data pipelines and efficiently leveraging large-scale datasets.
• Strong collaboration skills and the ability to work across teams and contribute to peers' success.
• Ability to thrive with a high level of autonomy and responsibility and an entrepreneurial mindset.

Responsibilities

• Design, build, train, evaluate, deploy, and own ML models in production that improve transaction categorization, risk scoring, and data enrichment across Financial Connections
• Design and build large-scale ML systems that operate on diverse financial data from thousands of institutions
• Experiment and iterate on ML models using PyTorch, TensorFlow, XGBoost to achieve data quality and accuracy goals
• Develop pipelines and automated processes to train and evaluate models in offline and online environments
• Integrate ML models into production systems and ensure scalability and reliability
• Collaborate with product, data science, and engineering partners across Stripe to identify ML opportunities for merchants and consumers
• Engage with latest ML/AI developments and translate ideas into production-ready solutions
• Mentor engineers and contribute to strong ML engineering culture within the team

Skills

ML deployment

Cross-team collaboration

Autonomy

Problem solving

Education

MS or PhD in ML/AI

Tools

PyTorch

TensorFlow

XGBoost

Spark

Job description

Stripe's Financial Connections team is hiring a senior ML engineer to design, train, and deploy production ML models at scale across thousands of financial institutions. You will own end-to-end ML pipelines, focusing on data quality, risk scoring, and enrichment for merchants and consumers, while collaborating with product, data science, and engineering partners.

You will mentor engineers and help grow a strong ML culture.