Skip to main content

Affirm

Machine Learning Engineer II (Servicing ML)

Remote US (Remote)$165k–$225kmidAdded 1 month ago

About this role

Affirm seeks a Machine Learning Engineer II to develop AI systems that automate customer servicing operations like disputes, chargebacks, and refunds. You'll build models using LLMs and tabular classification techniques, working across ML lifecycle tools and cross-functional teams to deploy production systems that improve customer experiences.

What you'll do

  • Develop AI systems to automate dispute and chargeback handling using structured evidence and business logic
  • Build models that accelerate refund processing and payment recovery for customers
  • Create and maintain evidence extraction pipelines using LLM-powered workflows to transform unstructured data into actionable outputs
  • Prototype modeling approaches, run offline experiments, and deploy high-performing solutions with risk controls
  • Collaborate with Engineering, Operations, Product, and ML Platform teams to define requirements and communicate results
  • Own model health and monitoring in production environments

What they're looking for

  • Python with production-quality code standards
  • Tabular classification modeling (LightGBM, XGBoost, CatBoost)
  • LLM API integration and prompt engineering (OpenAI, Anthropic)
  • ML lifecycle tooling (Kubeflow, Airflow, MLflow)
  • Document and unstructured data processing (PDF/image extraction, text parsing)
  • Orchestration frameworks (LangChain, LangGraph)
  • AI-powered developer tools (Claude, Cursor)
  • Code review and debugging in large codebases

Benefits

  • 100% subsidized medical, dental, and vision coverage for you and dependents
  • Flexible spending wallets for technology, food, lifestyle, and family planning
  • Equity compensation (Grade 6)
  • Remote-first work flexibility across the US
  • Competitive base pay: $165,000-$225,000 (CA/WA/NY/NJ/CT) or $146,000-$206,000 (other states)
  • Professional growth and development opportunities
Apply with Autofill

Opens the application — the Jobs AI extension fills it for you. Set up autofill

Opens the official application on the employer’s site. No login required.

Affirm

Affirm builds financial infrastructure and platforms that power point-of-sale lending, merchant operations, card issuance, and capital management at scale. The company is hiring backend engineers, AI solutions engineers, and infrastructure specialists to develop reliable, scalable systems for merchant platforms, payment processing, reliability monitoring, and capital orchestration.

Website
affirm.com
View all jobs at Affirm

Likely interview questions

  • Can you walk us through a production machine learning system you've built end-to-end? How did you handle model monitoring and what metrics did you track?
  • Describe your experience building tabular classification models with gradient-boosted trees. How did you approach feature engineering and model evaluation for your use case?