Lead Machine Learning Engineer

Bumbleinc; Austin (TX); On-site; USD 120,000 - 160,000; Full time

Benefits offered by this job

Medical/dental/vision insurance

Unlimited PTO

401k with company match

Paid parental leave

Life and long-term disability insurance

Job summary

Bumbleinc is hiring for an ML Engineer with responsibility for designing and delivering end-to-end machine learning systems to enhance recommendations and personalization. The ideal candidate will have 7-10 years of experience and strong expertise in machine learning, especially using frameworks like PyTorch.

This position offers a collaborative environment, where you will mentor others and drive experimentation to enhance product outcomes. Comprehensive benefits include unlimited PTO and competitive health insurance.

Qualifications

• Typically requires 7–10 years of experience.
• Strong expertise in machine learning in production.
• Experience designing and scaling ML pipelines and data systems.
• Demonstrated ability to take ownership of ambiguous problems.
• Hands‑on experience with LLMs.

Responsibilities

• Lead the design and delivery of ML systems.
• Architect and optimise ML pipelines.
• Build and deploy advanced models.
• Drive experimentation strategy.
• Mentor and support other engineers.

Skills

Machine learning systems

Python

ML frameworks (PyTorch, TensorFlow)

ML pipelines

LLMs

Collaboration skills

AI fluency

Tools

Spark

Airflow

Job description

Responsibilities

• Lead the design and delivery of end-to-end ML systems powering recommendations, ranking, and personalization, driving measurable improvements in user outcomes
• Architect and optimise ML pipelines, integrating data processing, model training, evaluation, and deployment into scalable systems (e.g. Spark, Airflow)
• Build and deploy advanced models using modern frameworks (e.g. PyTorch), ensuring robustness in high-scale production environments
• Develop and apply LLM-based capabilities, including prompt design and fine‑tuning approaches for production use cases
• Drive experimentation strategy, including A/B testing and offline evaluation, to continuously improve performance and inform product decisions
• Partner cross‑functionally with Product, Engineering, and Data teams, collaborating with purpose to translate business challenges into ML solutions
• Mentor and support other engineers, fostering a culture of Curiosity, ownership, and continuous improvement

Qualifications

• Typically requires 7–10 years of experience, though we welcome candidates with alternative backgrounds that demonstrate equivalent skills.
• Strong expertise in machine learning, with a track record of delivering end-to-end systems in production
• Proficiency in Python and ML frameworks (e.g. PyTorch, TensorFlow), with experience in areas such as recommendation systems, ranking, or NLP
• Experience designing and scaling ML pipelines and data systems (e.g. Spark, Airflow, or similar technologies)
• Hands‑on experience with LLMs, including prompting, evaluation, and production integration
• Demonstrated ability to take ownership of ambiguous problems and deliver impactful outcomes
• Strong collaboration skills, working cross‑functionally and taking ownership of shared goals
• Solid AI fluency, with the ability to guide best practices in model development and responsible AI usage within a team

Compensation

Will be determined based on factors such as the selected candidate’s qualifications, relevant experience, skill set, and other job‑related considerations.

Benefits & Perks

• Insurance: Medical/dental/vision, 30-day eligibility. Bumble has multiple competitive offerings that will be available to you on the first of the month following date of hire.
• Unlimited PTO + 1 company‑wide week off + Focus Fridays every week
• Fully paid life and long‑term disability insurance
• 401k with 4% company match if you contribute 6%, 90-day eligibility
• Monthly wellness benefit and access to Noom, Unmind, and Your Money Line
• Maternity and Fertility benefit + 26 week paid parental leave
• Premium App Access

Equal Opportunity Employer

Bumble Inc. is an equal opportunity employer and we strongly encourage people of all ages, colour, lesbian, gay, bisexual, transgender, queer and non‑binary people, veterans, parents, people with disabilities, and neuro‑divergent people to apply. We’re happy to make any reasonable adjustments that will help you feel more confident throughout the process, please don't hesitate to let us know how we can help.