Senior Staff Machine Learning Engineer, Post Training

Traveltechessentialist; United States; On-site; USD 248,000 - 310,000; Full time

Benefits offered by this job

Bonus

Equity

Employee Travel Credits

Job summary

Traveltechessentialist is seeking a skilled Principle Machine Learning Engineer to enhance AI-driven products. Located in the US, the role involves fine-tuning advanced LLMs and collaborating with cross-functional teams. These initiatives aim to improve customer experiences through innovative AI solutions. Ideal candidates should have a PhD and over 10 years in machine learning, proficient in Python and PyTorch. The position offers remote eligibility with competitive pay, ranging from $248,000 to $310,000, plus benefits.

Qualifications

• 10+ years experience with developing machine learning models and products at scale.
• Experience with frameworks such as PyTorch.
• Proven record of training and fine-tuning LLMs.

Responsibilities

• Fine-tune state-of-the-art LLMs for various use cases.
• Work with large-scale structured and unstructured data.
• Drive AI architectural decisions for products.

Skills

Machine Learning

Artificial Intelligence

Python

PyTorch

LLM fine-tuning

Education

PhD in Computer Science, Machine Learning, Mathematics, Statistics, or related technical field

Job description

Airbnb

Airbnb was born in 2007 when two hosts welcomed three guests to their San Francisco home, and has since grown to over 5 million hosts who have welcomed over 2 billion guest arrivals in almost every country across the globe. Every day, hosts offer unique stays and experiences that make it possible for guests to connect with communities in a more authentic way.

The Community You Will Join

Machine Learning and Artificial Intelligence are at the heart of the Airbnb product. From Trust to Payments, and from Customer Service to Marketing we rely on ML to ensure that guests and hosts have the best possible experience with Airbnb.

The CS AI product team is responsible for driving CSxAI (Customer Support x Artificial Intelligence) initiatives by adopting the Generative AI technologies to enable an intelligent, scalable and exceptional service experience. The team develops and enhances various AI models, ML services and tools including LLM fine‑tuning, alignment and optimization, RAG/Search, LLM evaluation and testing automation, feedback‑based learning and guardrail for a wide range of applications in Airbnb.

The Difference You Will Make

As a principle machine learning engineer, you will be responsible for fine‑tuning state‑of‑the‑art LLMs for diverse use cases while optimizing models for high‑performance deployment on Airbnb’s ML Infrastructure. You will partner with product managers, software engineers, data scientists and operation teams to brainstorm, design and develop AI products such as AI Assistant, Autonomous agent, recommendation, travel planning, and many more products that make meaningful impacts in the world of travel.

A Typical Day

• Work with large scale structured and unstructured data; explore, experiment, build, and continuously improve foundation models for Airbnb product, business and operational use cases.
• Create a multi‑year tech roadmap that enables our team to stay on the leading edge of the rapidly evolving AI landscape and leverage the best in class technologies to deliver customer benefits.
• Continuously evaluate recent and upcoming large foundational models, ensuring the selection and refinement of the highest quality models for enhanced performance and efficiency.
• Hands‑on prototype, develop and productionize LLM models and pipelines at scale, including both batch and real‑time use cases.
• Drive key AI architectural decisions for products, and contribute to Airbnb’s ML platform architecture and strategy.

Your Expertise

• PhD in Computer Science, Machine Learning, Mathematics, Statistics, or related technical field.
• 10+ years of experience with developing machine learning models and products at scale from inception to business impact.
• Programming experience in Python and hands‑on experience with frameworks such as PyTorch.
• Proven record of training, fine tuning, optimizing models and inference run‑time.
• Post‑training experience in areas like data processing for fine‑tuning; responsible LLMs; LLM alignment; reinforcement learning; efficient training and inference; language model evaluation; and/or multilingual and multimodal modeling.
• Or specialized experience in runtime optimizations, model quantization, compression, on‑device inference, GPU inference, PyTorch, kernel development.

Your Location

This position is US — Remote Eligible. The role may include occasional work at an Airbnb office or attendance at offsites, as agreed to with your manager. While the position is Remote Eligible, you must live in a state where Airbnb, Inc. has a registered entity. Click here for the up-to-date list of excluded states. This list is continuously evolving, so please check back with us if the state you live in is on the exclusion list. If your position is employed by another Airbnb entity, your recruiter will inform you what states you are eligible to work from.

Our Commitment To Inclusion & Belonging

Airbnb is committed to working with the broadest talent pool possible. We believe diverse ideas foster innovation and engagement, and allow us to attract creatively‑led people, and to develop the best products, services and solutions. All qualified individuals are encouraged to apply.

We strive to also provide a disability inclusive application and interview process. If you are a candidate with a disability and require reasonable accommodation in order to submit an application, please contact us at [e-mail]. Please include your full name, the role you’re applying for and the accommodation necessary to assist you with the recruiting process.

We ask that you only reach out to us if you are a candidate whose disability prevents you from being able to complete our online application.

How We'll Take Care Of You

Our job titles may span more than one career level. The actual base pay is dependent upon many factors, such as: training, transferable skills, work experience, business needs and market demands. The base pay range is subject to change and may be modified in the future. This role may also be eligible for bonus, equity, benefits, and Employee Travel Credits.

Pay Range

$248,000—$310,000 USD