Senior ML Engineer: LLM Fine-Tuning & Scalable AI (Remote)

Aimlroles; Northern (KY); Hybrid; USD 248,000 - 310,000; Full time

Job summary

Airbnb seeks a Senior Staff Machine Learning Engineer to fine-tune state-of-the-art LLMs and optimize deployments on Airbnb’s ML Infrastructure. You will partner with product managers, software engineers, data scientists and operations to design AI products such as AI Assistant, Autonomous agent, and travel planning.

You will work with large-scale data to build and improve foundation models, define a multi-year roadmap, and drive AI architectural decisions for scalable services across the

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.
• 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; multilingual/multimodal modeling.
• Specialized experience in runtime optimizations, model quantization, compression, on-device inference, GPU inference, pytorch, kernel development.

Responsibilities

• Fine-tune state-of-the-art LLMs for diverse use cases while optimizing models for high-performance deployment on Airbnb’s ML Infrastructure.
• 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 more.
• 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.

Skills

Python

PyTorch

Model training

Fine-tuning

Inference optimization

Education

PhD in Computer Science/Machine Learning/Math/Statistics

Tools

PyTorch

Job description

Airbnb seeks a Senior Staff Machine Learning Engineer to fine-tune state-of-the-art LLMs and optimize deployments on Airbnb’s ML Infrastructure. You will partner with product managers, software engineers, data scientists and operations to design AI products such as AI Assistant, Autonomous agent, and travel planning.

You will work with large-scale data to build and improve foundation models, define a multi-year roadmap, and drive AI architectural decisions for scalable services across the