Staff Machine Learning Engineer, Financial Connections

Job summary

Responsibilities

We're looking for machine learning engineers who want to build intelligent systems that provide financial data at scale. You'll play a key role in designing, training, and deploying ML models that improve the quality, accuracy, and usefulness of financial data across Stripe's ecosystem.

• 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 tools such as PyTorch, TensorFlow, XGBoost) to achieve key business goals around data quality and accuracy
• Develop pipelines and automated processes to train and evaluate models in offline and online environments
• Integrate ML models into production systems and ensure their scalability and reliability

What they're looking for

• 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

Tech stack

• PyTorch
• TensorFlow
• Spark
• Transformers