Affirm

Machine Learning Engineer II

Affirm
Apply
3 months ago
Remote, CanadaMid Level
H1B Sponsor

Base Salary

$125k - $175k/yr

Responsibilities

  • Develop AI systems that automate dispute and chargeback handling using structured evidence and business logic
  • Build models that automate refunds and accelerate customer reimbursements
  • Build and maintain evidence-extraction pipelines that process unstructured data through LLM-powered workflows
  • Prototype modeling approaches, run offline experiments, and move effective solutions into production with appropriate risk controls
  • Measure and monitor models and keep production systems healthy
  • Collaborate with Engineering, Servicing Operations, Product, and ML Platform to define requirements, evaluate tradeoffs, and communicate results

Requirements

  • At least 2 years of experience as a machine learning engineer
  • Strong Python skills and experience writing production-quality code
  • Experience building and evaluating tabular classification models, preferably with LightGBM, XGBoost, or CatBoost
  • Experience building applications with LLM APIs such as OpenAI or Anthropic, including structured extraction, prompt engineering, and orchestration with LangChain or LangGraph
  • Familiarity with document and unstructured-data processing, including PDF or image extraction and text parsing
  • Experience with ML lifecycle tooling for training orchestration, experimentation, and model monitoring, such as Kubeflow, Airflow, or MLflow
  • Proficiency using AI-powered developer tools such as Claude Code or Cursor
  • Ability to design solutions spanning multiple software components and write clear, well-tested, extensible code
  • Ability to navigate large codebases, debug others’ code, and provide code-review feedback
  • Strong verbal and written communication skills and a demonstrated commitment to professional growth

Benefits

  • Remote-first role with remote work available anywhere within Canada of employment
  • 100% subsidized medical, dental, and vision coverage for employees and dependents
  • Monthly or flexible spending stipends for technology, food, lifestyle, wellness, and family-forming expenses
  • Competitive vacation and holiday schedules
  • Employee stock purchase plan (ESPP)
  • Potential eligibility for equity rewards

Tech Stack

Apache AirflowLightGBMMLflowPythonXGBoost
Affirm

About Affirm

1,001-5,000 employees

At Affirm, we’re using today’s technology to bring significant disruptive innovation to the financial industry. We focus on improving the lives of consumers by delivering simple, honest and transparent financial products. We have roles across a variety of teams, including engineering, product, finance, operations, revenue, marketing, people and more. A note to applicants: Fake job and recruiting scams are on the rise. Affirm will never ask you to pay any upfront fees or send you any checks to purchase equipment during any part of the recruiting process. Affirm only uses legitimate company email address domains (either “@affirm.com” or "@us.greenhouse-mail.io”) to communicate with candidates via email. Affirm is also proud to be a remote-first company! Attracting great people starts by finding talent in different places. That's why we built remote work into the DNA of how we do business—but if an office space is more your style, we have offices in Chicago, New York City, Pittsburgh, San Francisco, Seattle, Toronto, Warsaw, Madrid and London. Not to mention we have comprehensive and award-winning benefits that were made for you. Interested in learning more? Check out our careers page and apply today! Visit affirm.com/careers for more information.