Staff Software Engineer, Machine Learning Infrastructure

San Francisco, Remote

Applications have closed
Airbnb logo

Posted 8 months ago

At Airbnb, our mission is to create a world where anyone can belong anywhere. We use Machine Learning extensively to create a more connected, empowered, and safer global community and enable the next generation of intelligent, worry-free travel experience.

In this role, you’ll help us build Airbnb’s end-to-end ML Platform - scalable shared infrastructure that accelerates the pace of Machine Learning development and the deployment of impactful, high quality ML use cases company-wide. Our ML platform spans the entire ML lifecycle from labeling, prototyping, training at scale, offline experimentation, serving & lifecycle management and observability and quality upkeep.

You’ll have the opportunity to:

  • Build Model and Feature Serving infrastructure that can handle 10-100x our current scale to meet the evolving needs of the business
  • Build Zipline, our industry-leading feature engineering platform that enables engineers and data scientists to rapidly prototype and productionize features with high quality
  • Build out distributed training & model experimentation systems to enable rapid iteration
  • Create intelligent ML and Data aware real-time monitoring & observability systems
  • Enable ML use cases all across Airbnb’s product - for example Search Ranking, Fraud Detection, Customer Support and many more!


  • Work closely with Machine Learning Engineers and Data Scientists to understand, refine, and prioritize requirements
  • Design and create simple, powerful APIs (& UIs) to capture our user needs
  • Design, build and scale Distributed backend services built on Docker and Kubernetes
  • Design, build and scale Real-Time and Batch processing pipelines leveraging Spark & Flink

Who we are looking for:

  • 8+ years of industry experience (and/or relevant academic experience)
  • Strong coding skills in Python/Java/Scala or equivalent
  • Solid understanding of engineering and infrastructure best practices
  • Experience developing and productionizing machine learning models is a plus
  • Experience with Docker, Kubernetes, Hive, or Spark is a plus
  • Industry experience building end-to-end Machine Learning Infrastructure a big plus


  • Stock
  • Competitive salaries
  • Quarterly employee travel coupon
  • Paid time off
  • Medical, dental, & vision insurance
  • Life insurance and disability benefits
  • Fitness discounts
  • 401K
  • Flexible Spending Accounts
  • Apple equipment
  • Commuter subsidies
  • Community involvement (4 hours per month to give back to the community)
  • Company sponsored tech talks and happy hours
  • Breakfast, lunch, and dinner
  • Much more... 

We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status

Job tags: Engineering Java Kubernetes Machine Learning ML Python Scala Spark Travel
Job region(s): North America Remote/Anywhere
Job stats:  40  6  0

More AI/ML/Data Science position highlights