Skip to content
StripeStripeUnited States

Machine Learning Engineer, Capital Underwriting

Build, train, and deploy production ML models for underwriting and portfolio management at Stripe Capital. Requires 5+ years shipping ML systems with PyTorch/TensorFlow and experience in lending/fraud domains.

Salary not listed
Remote5+ YOEML Engineering

About the role

Responsibilities

  • Design state-of-the-art ML models and large scale ML systems for underwriting and portfolio management for Stripe Capital based on ML principles, domain knowledge, risk, regulatory and engineering constraints
  • Design systems to speed up the time from idea to deployment of new models
  • Experiment and iterate on ML models (using tools such as PyTorch and TensorFlow) to achieve key business goals and drive efficiency
  • 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 and strategy partners to propose, prioritize, and implement new product features
  • Engage with the latest developments in ML/AI and take calculated risks in transforming innovative ML ideas into productionized solutions

Requirements

  • 5+ 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
  • Knowledge of various ML algorithms and model architectures
  • 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 complicated data pipelines and efficiently leveraging large-scale datasets
  • Hands-on experience in collaborating across multiple teams, especially Data Science and Risk Management teams

Preferred Qualifications

  • MS/PhD degree in ML/AI or related field (e.g. math, physics, statistics)
  • Proven track record of building and deploying ML systems that have effectively solved ambiguous business problems
  • Experience in adversarial domains such as Lending, Trading, Fraud
  • Experience with Deep Learning including the latest architectures such as transformers, test-time compute, reinforcement learning

Skills

PyTorchTensorFlowXgboostSparkMachine LearningDeep LearningTransformersReinforcement LearningData PipelinesModel Deployment

Similar roles

ML Engineering jobs
Cerebras Systems

CoDesign & NextGen Performance Engineer

Cerebras SystemsSunnyvale, CA

Characterize, analyze, and optimize performance of state-of-the-art AI models on Cerebras' wafer-scale hardware. Build performance models, optimize kernels and compilers, debug runtime behavior, and develop visualization tools to influence next-gen AI architecture.

Salary not listed
On-site3+ YOEML Engineering
OpenAI

Research Engineer, Privacy

OpenAISan Francisco, CA

Research Engineer on OpenAI's Privacy team designing and prototyping privacy-preserving ML algorithms like differential privacy and federated learning at scale. Requires hands-on PETs experience, fluency in PyTorch/JAX, and a track record implementing or publishing novel privacy work.

380k – 445k/yr
HybridML Engineering
Console

Research Engineer

ConsoleSan Francisco, CA

Research Engineer building self-improving AI agent systems at Console. Develop eval/optimization loops, fine-tune specialist models, and improve agent reasoning over enterprise context using production data to drive measurable gains in quality, latency, and reliability.

200k – 350k/yr
On-siteML Engineering
Notion

Software Engineer, AI Platform

NotionSan Francisco, CA +1

Build and scale the shared AI platform foundations at Notion, enabling fast and safe shipping of AI products. Requires experience with LLM/ML platforms, strong ownership, and comfort across backend, infrastructure, and product code.

180k – 201k/yr
Hybrid5+ YOEML Engineering
Liftoff

Machine Learning Engineer

LiftoffCalifornia

Machine Learning Engineer building statistical models, optimization systems, and experiments for mobile ad tech economics on the Revenue Engine team. Requires PhD in CS/ML/Economics and industry experience applying ML or economics at scale.

215k – 275k/yr
RemoteML Engineering