Staff ML Engineer — Open Banking Data & Signals

Triwill Group; New York (NY); On-site; USD 180,000 - 230,000; Full time

Job summary

Stripe’s Financial Connections team is seeking a senior ML engineer to design, train, and deploy production-ready models that enhance financial data quality and usefulness across our platform. You will build large-scale systems handling data from thousands of institutions and collaborate with product and data science partners to deliver impactful merchant and consumer outcomes.

You will lead end-to-end ML initiatives, iterate on model architectures, and ensure reliable scalability in both

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.

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 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
• Collaborate with product, data science, and engineering partners across Stripe to identify opportunities where ML can improve outcomes for merchants and consumers
• Engage with the latest ML/AI developments and take calculated risks in transforming innovative ideas into productionized solutions
• Mentor engineers and contribute to a strong ML engineering culture within the team

Skills

ML systems in production

PyTorch/TensorFlow/XGBoost

Spark

Data pipelines

Autonomy

Cross-team collaboration

Education

MS or PhD in ML/AI

Tools

PyTorch

TensorFlow

XGBoost

Spark

Job description

Stripe’s Financial Connections team is seeking a senior ML engineer to design, train, and deploy production-ready models that enhance financial data quality and usefulness across our platform. You will build large-scale systems handling data from thousands of institutions and collaborate with product and data science partners to deliver impactful merchant and consumer outcomes.

You will lead end-to-end ML initiatives, iterate on model architectures, and ensure reliable scalability in both