Machine Learning Engineer II (Underwriting ML)

RemotemidPosted Aug 17, 2026

PythonMachine learningPyTorchSparkXGBoostKubeflowAirflowMLflow+13 more

About this role

Build and improve ML systems for real-time transaction decisions at Affirm's checkout, assessing repayment risk and expected value. Work on tabular and sequential data models from prototype to production with strong monitoring and measurement.

Must have

2+ years of experience as a machine learning engineer or PhD in relevant field

Strong Python skills and production-quality code experience

Experience building classification models with gradient-boosted decision trees

Experience with deep learning framework PyTorch preferred

Experience with distributed data processing frameworks Spark preferred

Experience with ML lifecycle tooling for training orchestration and monitoring

Proficient in using AI-powered developer tools for day-to-day workflows

Bachelor's degree in related field or equivalent practical experience

Technologies

PythonAWSSparkGitDockerRESTLinuxMachine learningJSONAirflowData analyticsPyTorchAIMLflowKubeflowRayCI/CDXGBoostLLMsLightGBMCatBoost

Responsibilities

Develop and iterate on underwriting prediction models for tabular and sequential data

Build and scale feature pipelines and training datasets from proprietary and third-party signals

Prototype new modeling ideas, run offline experiments, and drive best approaches into production

Productionize models into batch and real-time decision systems with risk controls

Instrument and monitor model and data health, define retraining and backtesting workflows

Collaborate across Engineering, Risk Analytics, Product, and ML Platform teams

Benefits

Flexible time offHealth insuranceDental insuranceVision insuranceEmployee share purchase planWellness platform with incentivesCompetitive salary and stock option planHome office allowance

Amenities

100% subsidized medical, dental and vision coverage for you and dependentsFlexible Spending Wallets for technology, food, lifestyle, and family forming expensesCompetitive vacation and holiday schedulesESPP enabling share purchases at a discount