Senior Staff Machine Learning Engineer, Post Training

About the Role

As a Senior Staff Machine Learning Engineer on the CS AI product team, 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 design and develop AI products such as AI Assistants, autonomous agents, recommendations, and travel planning.

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 enabling the team to stay on the leading edge of AI and leverage best-in-class technologies.
• Continuously evaluate recent and upcoming large foundational models, ensuring selection and refinement of high-quality models for performance and efficiency.
• Hands-on prototype, develop, and productionize LLM models and pipelines at scale, covering 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 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 run-time.
• 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/multimodal modeling; OR specialized experience in runtime optimizations, model quantization, compression, on-device inference, GPU inference, PyTorch, and kernel development.