RemoteLeaf Logo
Log in Sign up

Machine Learning Engineer II

2 months, 2 weeks ago
United States
Full-time
Junior
Machine Learning Engineer
Software Development
Apache Airflow CatBoost Kubeflow LightGBM Machine Learning MLflow Python XGBoost
Apply Now
Affirm

Affirm

Affirm offers a transparent buy now, pay later service founded in 2012 by Max Levchin. No late fees or surprises, just a responsible way to pay over time for your favorite brands.

Diversified Financial Services
1K-5K
Founded 2012
View All Jobs 101

Description

  • Develop AI systems that automate dispute and chargeback handling using structured evidence and business logic.
  • Build models that automate refunds and help customers receive money back faster.
  • Build and maintain evidence extraction pipelines that process unstructured data using LLM-powered workflows.
  • Prototype new modeling approaches, run offline experiments, and push the best-performing solutions into production with appropriate risk controls.
  • Collaborate with Engineering, Servicing Operations, Product, and ML Platform to define requirements and evaluate tradeoffs.
  • Communicate results clearly to technical and non-technical stakeholders.
  • Monitor and maintain model health using strong measurement and monitoring practices.
  • Work closely with ML engineers and platform partners to take models from idea to production.

Requirements

  • 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 using gradient-boosted decision trees such as LightGBM, XGBoost, or CatBoost.
  • Experience building applications with LLM APIs such as OpenAI or Anthropic, including structured extraction, prompt engineering, and orchestration frameworks like LangChain or LangGraph.
  • Familiarity with document and unstructured data processing, such as PDF/image extraction or text parsing.
  • Experience with ML lifecycle tooling for training orchestration, experimentation, and model monitoring, such as Kubeflow, Airflow, MLflow, or equivalent platforms.
  • Proficiency using AI-powered developer tools such as Claude Code or Cursor to accelerate iteration, debugging, and code quality.
  • Ability to design solutions that involve multiple software components and implement them in clear, well-tested, extensible code.
  • Comfort navigating a large codebase, debugging others' code, and giving feedback through code reviews.
  • Strong verbal and written communication skills.
  • Equivalent practical experience or a Bachelor’s degree in a related field.

Benefits

  • Base salary range of $160,000-$210,000 for CA, WA, NY, NJ, and CT, or $142,000-$192,000 for other U.S. states.
  • Equity rewards as part of the total compensation package.
  • Monthly stipends for health, wellness, and tech spending.
  • 100% subsidized medical coverage for you and your dependents.
  • Dental and vision coverage for you and your dependents.
  • Remote-first work environment with the flexibility to work almost anywhere within the country of employment.
  • Competitive vacation and holiday schedules.
  • Employee stock purchase plan with a discount on Affirm shares.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Lyra Health

Staff ML Engineer (ML/AI)

Lyra Health 1K-5K Health Care Providers & Services

Lyra Health is hiring a Staff ML/AI Engineer to define the architecture and technical strategy for enterprise-scale machine learning and generative AI platforms supporting clinically precise, reliable mental health products.

United States Full-time Lead Machine Learning Engineer
$161k-$222k
AWS Celery CI/CD Docker Generative AI HIPAA Java Kafka Kotlin Kubernetes Machine Learning Microservices MLOps Python REST API
1 day, 17 hours ago
Boundless

Applied AI/ML Engineer

Boundless Internet Software & Services

Boundless is hiring an Applied AI/ML Engineer to build and ship end-to-end AI products on its GPU inference fleet, from low-latency serving through reinforcement-learning post-training and production evaluation.

United States Full-time Mid Level Machine Learning Engineer
$175k-$250k
Docker Kubernetes Python PyTorch Reinforcement Learning
1 day, 17 hours ago
Lyra Health

Sr. ML Engineer (MLOps)

Lyra Health 1K-5K Health Care Providers & Services

Lyra Health is hiring a Senior ML Engineer to build and operate the tooling and services behind production machine learning and generative AI products for its mental health platform.

United States Full-time Senior Machine Learning Engineer
$143k-$197k
AWS Celery Docker Generative AI Java Kotlin Kubernetes Machine Learning MLOps Python REST API
1 day, 17 hours ago
DEFCON AI

Data & ML Engineer

DEFCON AI 11-50 Internet Software & Services

Defcon AI is hiring a Data & ML Engineer to build the data, matching, scoring, and retrieval layers of an AI-enabled decision-support system in a controlled government cloud environment.

United States Full-time Senior Machine Learning Engineer
$150k-$200k
Apache Airflow Apache Spark dbt DevSecOps Kafka Machine Learning PostgreSQL Python PyTorch Scikit-learn SQL XGBoost
2 days, 16 hours ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers