Machine Learning Engineer II (fraud)

Affirm Affirm · Fintech · Canada · Remote · Checkout

Machine Learning Engineer II focused on fraud detection at Affirm. This role involves developing and iterating on fraud prediction models, building and scaling feature pipelines, prototyping new modeling ideas, productionizing models into real-time decision systems, and monitoring model health. Requires experience with tabular data, gradient-boosted trees, deep learning frameworks, distributed data processing, and ML lifecycle tooling.

What you'd actually do

  1. You will develop and iterate on fraud prediction models using a mix of approaches for tabular and behavioral data
  2. You will build and scale feature pipelines and training datasets from proprietary and third-party signals, partnering with data and platform teams when needed.
  3. You will prototype new modeling ideas and features, run offline experiments, and drive the best-performing approaches into production with appropriate risk controls.
  4. You will help productionize models: integrate into batch and/or real-time decision systems, and improve reliability, latency, and operational robustness.
  5. You will instrument and monitor model and data health, and help define retraining/backtesting workflows as fraud patterns evolve.

Skills

Required

  • 2+ years of experience as a machine learning engineer or a PhD in a relevant field
  • Strong Python skills
  • Experience building and evaluating models for tabular classification problems
  • Experience with a deep learning framework (PyTorch preferred)
  • Experience working with distributed data processing or parallel compute frameworks (Spark preferred)
  • Experience with ML lifecycle tooling for training orchestration, experimentation, and model monitoring

Nice to have

  • LightGBM/XGBoost/CatBoost
  • Ray/Dask
  • Kubeflow, Airflow, MLflow, or equivalent internal platforms
  • AI-powered developer tools (e.g., Claude Code, Cursor, or similar)

What the JD emphasized

  • production-quality code
  • production
  • productionize models
  • operational robustness

Other signals

  • build and improve machine learning systems that make real-time transaction decisions
  • take models from idea to prototype to production
  • keep them healthy with strong measurement and monitoring