Machine Learning Engineer II (Servicing ML)

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