Skip to content
Bumble logo

Staff Machine Learning Engineer

BumbleOnline Dating company
Austin, United StatesLead
Data & AI

About the role

Lead ML Engineer to drive end-to-end ML systems, focusing on recommendations, personalization, and LLM-powered features.

  • Bumble's Machine Learning team shapes how millions connect, build trust, and find belonging by delivering intelligent systems that elevate user experience and safety.
  • This role operates at the intersection of product innovation and technical excellence.
  • Key Responsibilities Own and drive large-scale ML systems for recommendations, ranking, and personalization.
  • Design and architect scalable ML platforms and pipelines using Spark and Airflow.
  • Build and optimize advanced models using PyTorch, including LLM-powered systems.
  • Establish best practices for experimentation and A/B testing.
  • Mentor engineers and foster a culture of technical excellence.
  • Requirements 8-12 years of experience in machine learning and scaling production systems.
  • Strong proficiency in Python and ML frameworks like PyTorch.
  • Expertise in designing scalable ML and data pipelines using distributed systems (Spark, Airflow).
View original posting →

Tech stack

PythonSparkAirflowPyTorchLLMs

Match insights

Tech:Python, Spark, Airflow, PyTorch, LLMs
Level:Lead

More roles at Bumble

View open roles at Bumble