Senior ML Engineer - Personalization & Scalable Pipelines

Bumble; Austin, Northern (TX, KY); Hybrid; USD 150,000 - 210,000; Full time

Job summary

Bumble Inc. is seeking an experienced machine learning professional to build and deploy models that enhance recommendations, ranking, and personalization for users.

You will own projects end-to-end, from data exploration to production deployment, and develop scalable ML pipelines using Spark and Airflow. Proficiency in Python and at least one major ML framework is required, with a strong background in experimentation and cross-functional collaboration.

Qualifications

• Typically requires 5-8 years of experience.
• Strong experience building and deploying ML models in production environments.
• Proficiency in Python and experience with at least one major ML framework (e.g. PyTorch, TensorFlow).
• Experience working with data pipelines and distributed systems (e.g. Spark, Airflow).
• Familiarity with experimentation methodologies such as A/B testing and model evaluation techniques.
• Ability to collaborate effectively across functions, showing ownership and collaboration.
• Demonstrates an agile mindset, adapting approaches based on data and priorities.
• Growing AI fluency, applying ML techniques and emerging tools responsibly.

Responsibilities

• Build and deploy machine learning models that improve recommendations, ranking, and personalization, driving measurable impact on user experience and engagement
• Own problems end-to-end, from data exploration and feature engineering through to model training, evaluation, and production deployment
• Develop and maintain scalable ML pipelines using tools such as Spark and Airflow to support reliable, high-quality model delivery
• Apply modern ML frameworks (e.g. PyTorch or TensorFlow) to design, train, and optimize models in production environments
• Contribute to experimentation frameworks, including A/B testing and offline evaluation, to iterate on model performance with an agile mindset
• Collaborate cross-functionally with Product and Engineering, translating product questions into ML solutions
• Take ownership of delivering high-quality solutions and see work through from insight to impact, balancing speed and rigor
• Apply responsible AI practices, ensuring fairness, transparency, and safety are considered in model development and deployment

Skills

Python

ML frameworks

Distributed systems

Experimentation

Ownership & collaboration

Tools

Spark

Airflow

PyTorch

TensorFlow

Job description

Bumble Inc. is seeking an experienced machine learning professional to build and deploy models that enhance recommendations, ranking, and personalization for users.

You will own projects end-to-end, from data exploration to production deployment, and develop scalable ML pipelines using Spark and Airflow. Proficiency in Python and at least one major ML framework is required, with a strong background in experimentation and cross-functional collaboration.