Jobs · Engineering

Principal AI/ML Researcher / Engineer In Reasoning, Planning, and Decision-making systems

Airbnb · United States · 1 wk ago
RemoteRemoteEngineering$296k–$370k/yrFull-time

Job Summary

We are seeking a Principal / Distinguished AI/ML Researcher and/or Engineer with deep experience in reasoning, planning, and decision-making systems. This role is ideal for individuals who have architected post-training intelligence frameworks, integrated Large Reasoning Models (LRMs) with Knowledge Graphs, and applied Reinforcement Learning (RL) as a first-class component of adaptive planning and control.

About the Role

This role advances Airbnb's AI capabilities toward reasoning, planning, and adaptive decision-making across complex real-world environments. The near-term impact spans improved decision quality, contextual intelligence, adaptive personalization, and operational coordination across guest and host workflows — introducing goal-directed reasoning systems capable of handling ambiguity, constraints, trade-offs, and multi-step planning. Guests benefit from more intelligent planning and assistance, while hosts and internal teams gain systems capable of adaptive optimization, dynamic recommendations, policy-aware decisioning, and intelligent workflow orchestration.

Responsibilities

  • Drive foundational and applied research in reasoning engines, planning architectures, and decision-making frameworks at scale in order to incorporate genAI into the ranking / recommendation / personalization stack in both single model to multi-agent (system) level intelligence with objective to grow the business (new user growth, abandoned user, long tailed user) in existing and new business areas while supporting Multi-Modal NL → Conversational Interfaces.
  • Advance techniques in LLM/LRM post-training, reinforcement learning–based decisioning, and knowledge-integrated agents.
  • Design methods for plan induction, value estimation, and contingency modeling within intelligent agents.
  • Explore and validate protocols for distributed reasoning and joint planning among cooperative agents in multi-agent systems.
  • Architect RPD systems that integrate post-trained LLMs/LRMs, graph-structured memory (e.g., KGs), and RL-driven controllers.
  • Design recursive task planners, search-based or policy-based reasoners, and belief-state trackers that can interoperate with large model substrates.
  • Ensure modularity and extensibility through multi-agent frameworks, agentic substrates, and declarative planning pipelines.
  • Define communication protocols, coordination strategies, and cross-agent knowledge alignment mechanisms to foster emergent cooperative intelligence.
  • Build and evolve stateful, dynamic models that combine supervised learning with online/offline reinforcement, simulation-based rollouts, and symbol grounding.
  • Implement hybrid pipelines that couple learned embeddings, prompted generative models, and graph-theoretic inference.
  • Optimize systems for adaptive exploration, planning horizon control, and policy robustness.
  • Develop frameworks for distributed value propagation, multi-agent credit assignment, and global planning from local agents.
  • Set direction for planning/reasoning infrastructure within the AI/ML platform strategy.
  • Serve as the technical conscience and architectural leader across high-stakes AI initiatives involving autonomous agents or high-fidelity decision pipelines.
  • Mentor teams in systems thinking, causal modeling, symbolic-connectionist integrations, and long-term planning under uncertainty.
  • Lead development of multi-agent reasoning systems, defining principles for inter-agent knowledge exchange, goal delegation, and cooperative decision resolution.
  • Work across disciplines—product, infra, and design—to translate ambiguous product intent into multi-stage reasoning pipelines.
  • Partner with researchers, ontologists, and ML engineers to encode world knowledge, goals, and values into usable inference artifacts.
  • Contribute to a company-wide understanding of what it means to make intelligent choices, not just predictions.
  • Collaborate with internal teams on distributed agent coordination, shared memory protocols, and policy harmonization across decision surfaces.
  • Productionize real-time reasoning loops with low-latency inference, caching, retrieval-augmented generation, and streaming updates to symbolic memory.
  • Deploy post-training hooks for inserting logic, constraints, and domain priors into existing large models.
  • Create advanced monitoring, attribution, and evaluation pipelines for agent behavior and decision quality.
  • Operationalize multi-agent orchestration, ensuring reliable and fault-tolerant communication and decision propagation.

Requirements

  • Masters or equivalent in Computer Science, AI, Cognitive Science, or related fields.
  • Recent published work or patents in AI, Cognitive Science, or related fields.
  • 15+ years in AI/ML, including post-training architectures and production-scale reasoning systems.
  • Advanced coding proficiency in Java, Python, C++, or similar, with experience in ML/RL frameworks (e.g., PyTorch, Ray, JAX, RLlib) at scale.
  • Proven experience integrating LLMs/LRMs with Knowledge Graphs or structured world models.
  • Deep understanding of Reinforcement Learning and its application to decisioning and planning.
  • Fluency in hybrid model architectures: connectionist-symbolic fusion, retrieval-based agents, or goal-directed transformers.
  • Experience working on multi-agent coordination, distributed RL, or cooperative inference systems.
  • Ph.D. in AI, Machine Learning, Robotics, Cognitive Systems, or related areas.
  • Published work or patents in multi-agent reasoning, plan synthesis, knowledge-augmented learning, or generative control.
  • Experience in cognitive architectures, neuro-symbolic systems, or agent-based simulation environments.
  • Demonstrated ability to lead cross-functional research-to-production transitions.
  • Experience with memory architectures, task graphs, or semantic program induction.
  • Prior work on distributed intelligence platforms with explicit agent interaction models and collective decision-making logic.

Qualifications

  • Experience in cognitive architectures, neuro-symbolic systems, or agent-based simulation environments.
  • Prior work on distributed intelligence platforms with explicit agent interaction models and collective decision-making logic.

Benefits

  • Flexible work arrangements.
  • Competitive compensation package.
  • Health, dental, and vision insurance.
  • Retirement savings plans.
  • Employee travel credits.

Pay

$296,000—$370,000 USD

Schedule

Remote Eligible

Similar jobs

Principal AI/ML Engineer

Sierra Nevada CorporationLone Tree, CO· 2 mo ago
Engineering$165k–$227k/yrapply on snc.wd1.myworkdayjobs.com