Principal Machine Learning Engineer- LLM Fine-Tuning And Optimization
Used Tools & Technologies
GenAIRequired Skills & Competences
Tag name is followed by "@" symbol and proficiency level value.
About proficiency levels:
- 1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;
- 3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;
- 7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;
- 10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.
Marketing @ 4
Python @ 4
Statistics @ 4
Machine Learning @ 4
Data Science @ 4
Mathematics @ 4
Customer Support @ 4
LLM @ 4
PyTorch @ 4
Agile @ 4
GPU @ 4
Generative AI @ 4
AI @ 4
Reinforcement Learning @ 4
RAG @ 4
- 1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;
- 3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;
- 7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;
- 10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.
Details
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 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 guardrails for a wide range of applications in Airbnb.
What you will do
As a principal 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 operations teams to brainstorm, design and develop AI products such as AI Assistant, autonomous agents, recommendation systems, travel planning, and other products.
Responsibilities
- 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 technical roadmap to keep the team on the leading edge of the evolving AI landscape and leverage best-in-class technologies.
- Continuously evaluate recent and upcoming large foundational models and select/refine models for 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.
Requirements
- PhD in Computer Science, Machine Learning, Mathematics, Statistics, or a related technical field.
- 10+ years of experience 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, and optimizing models and inference runtime.
- Post-training experience in areas such as 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, and kernel development.
Preferred Qualifications
- PhD in AI, machine learning, data science, or related technical fields (listed again as preferred).
- Publications at peer-reviewed AI conferences (e.g., NeurIPS, CVPR, ICML, ICLR, ICCV, ACL).
- Experience with AI technologies applied to customer support systems.
- Experience with the full AI product development lifecycle from incubation to production at scale, following agile practices in Applied AI/ML.
- Experience deploying and scaling business-critical AI services and driving architectural requirements on ML infrastructures.
Your Location
This position is US - Remote Eligible. The role may include occasional work at an Airbnb office or attendance at offsites, as agreed with your manager. While Remote Eligible, you must live in a state where Airbnb, Inc. has a registered entity (see Airbnb careers for excluded states). If employed by another Airbnb entity, your recruiter will inform you of eligible states.
How We'll Take Care Of You
Base pay depends on factors such as training, transferable skills, work experience, business needs and market demands. This role may also be eligible for bonus, equity, benefits, and Employee Travel Credits.
Pay Range
$292,000—$365,000 USD
Offices
United States