Principal Machine Learning Engineer- LLM Fine-tuning and Optimization

Job summary

Responsibilities

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.

• Work with large scale structured and unstructured data; explore, experiment, build and continuously improve foundation models for Airbnb product, business and operational use…
• 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…
• 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.

What they're looking for

• 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…

Tech stack

• RAG
• Python
• PyTorch