Machine Learning Engineer, Supportability

Stripe Stripe · Fintech · Canada · 8217 Risk Engineering

Stripe is seeking a Machine Learning Engineer for their Supportability Evaluation team. This role focuses on designing, building, training, evaluating, and deploying AI/ML models and large-scale systems for detection and decisioning within Stripe's financial ecosystem. The engineer will work on scaling an LLM-based system, integrating new capabilities through agentic approaches or supervised learning, and ensuring merchant compliance in real-time.

What you'd actually do

  1. Design state-of-the-art AI/ML models and large scale systems for detection and decisioning for Stripe products based onAI/ML principles, domain knowledge, and engineering constraints
  2. Drive the expansion of Stripe's largest LLM-based system, scaling its usage and integrating new capabilities through agentic approaches or supervised learning.
  3. Rapidly prototype new AI/ML-based approaches to achieve key business goals.
  4. Develop processes to train and evaluate models in offline and online environments
  5. Integrate models into production systems and ensure their scalability and reliability

Skills

Required

  • Proficient with AI/ML libraries and frameworks such as PyTorch, TensorFlow, XGBoost, as well as Spark
  • Knowledge of various AI/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
  • Experience rigorously evaluating model performance, including cleaning data, and working with data-generating processes to improve signal and reduce noise in high-noise datasets.
  • Proficiency in creatively applying modern machine learning techniques and Generative AI models to solve complex business problems.

Nice to have

  • MS/PhD degree in AI/ML or related field (e.g. math, physics, statistics)
  • Experience with DNNs including the latest architectures such as transformers and LLMs
  • Experience working in Java or Ruby codebases
  • Proven track record of building and deploying AI/ML systems that have effectively solved ambiguous business problems
  • Experience with online experimentation such as A/B testing or multi-armed bandits.
  • Experience with model calibration

What the JD emphasized

  • AI/ML models and systems
  • detect and action supportability violations in real-time
  • building high-fidelity detection engines
  • ensure our merchants remain compliant across the globe
  • shipping AI/ML systems in production
  • productionizing and deploying models at scale
  • rigorously evaluating model performance
  • applying modern machine learning techniques and Generative AI models to solve complex business problems

Other signals

  • AI/ML models and systems
  • detection and actioning supportability violations
  • high-fidelity detection engines
  • merchant compliance
  • LLM-based system
  • agentic approaches