Principal Machine Learning Engineer- LLM Fine-tuning and Optimization
About the role
Principal ML engineer focused on fine-tuning state-of-the-art LLMs and optimizing models for high-performance deployment on Airbnb’s ML infrastructure, collaborating with product, software, data science, and operations to deliver AI products and scalable AI services.
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 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.
Qualifications
- 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 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
Skills
Benefits
- Base pay range: $292,000—$365,000 USD
- Bonus, equity, benefits, and Employee Travel Credits eligibility
- Discretionary benefits as described in company policy
About Airbnb
Book accommodations around the world.
Founded in August of 2008 and based in San Francisco, California, Airbnb is a trusted community marketplace for people to list, discover, and book unique accommodations around the world — online or from a mobile phone. Whether an apartment for a night, a castle for a week, or a villa for a month, Airbnb connects people to unique travel experiences, at any price point, in more than 33,000 cities and 192 countries. And with world-class customer service and a growing community of users, Airbnb is the easiest way for people to monetize their extra space and showcase it to an audience of millions. No global movement springs from individuals. It takes an entire team united behind something big. Together, we work hard, we laugh a lot, we brainstorm nonstop, we use hundreds of Post-Its a week, and we give the best high-fives in town. Headquartered in San Francisco, we have satellite offices in Dublin, London, Barcelona, Paris, Milan, Copenhagen, Berlin, Moscow, São Paolo, Sydney, and Singapore.
More open roles at Airbnb
- Senior Software Engineer, Streaming Infrastructure4d agoUnited States
- Staff Software Engineer - Hosting Services4d agoHybrid (London, United Kingdom) · Hosting Services
- Staff Data Engineer, Enterprise Core Datasets, Tech Foundations4d agoSan Francisco, CA · Tech Foundations
- Senior Support Specialist, Premium4d agoGurugram · Customer Support / Premium Support
- Software Engineer, Payments4d agoHybrid (US - Remote Eligible) · Payments