Senior Staff Machine Learning Engineer, Post Training
Senior Staff ML Engineer fine-tunes and optimizes state-of-the-art LLMs for Airbnb's customer support AI products, including AI assistants and autonomous agents. Partners cross-functionally to productionize models at scale. Requires PhD and 10+ years experience with PyTorch.
About the job
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.
Requirements
- 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 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
PyTorch, Python, Llm Fine-Tuning, Llm Alignment, Reinforcement Learning, Model Optimization, Model Quantization, Gpu Inference, Multimodal Modeling, Language Model Evaluation
Similar jobs
ML Engineering jobs
Haus
Seattle, WAStaff Machine Learning Engineer building and operating production ML systems for causal marketing measurement, optimization, and planning. The role requires deep statistical and machine learning expertise, production programming experience, cross-functional collaboration, and technical mentorship.
Scale AI
San Francisco, CAStaff engineer responsible for designing and scaling the infrastructure, execution environments, verifiers, and tooling used to train and evaluate AI agents. Requires 8+ years of software engineering experience, strong Python and distributed-systems expertise, and familiarity with sandboxing, high-throughput systems, and LLM workflows.
United States
Leads the technical direction of large-scale ML infrastructure for embedding, recommendation, and personalization systems. The role requires 8+ years of ML engineering experience, expertise in deep learning and distributed training, and strong leadership across research, infrastructure, and production deployment.
Snowflake
Bellevue, WA
Leads the roadmap and technical vision for Snowflake Feature Store, building reliable, high-performance machine learning platform capabilities and supporting technical execution across partner teams. Requires 10+ years of experience with data-serving infrastructure or ML platforms, plus Java and Python expertise.
Shield AI
San Mateo, CA
Leads the design, implementation, integration, and field validation of tactical autonomy and multi-agent coordination capabilities for unmanned platforms. Requires 7+ years of relevant experience, production C++, technical leadership, and eligibility for a U.S. Secret clearance.