Staff Software Engineer, AI Serving Infrastructure

Location:
This role is completely remote-friendly. If you happen to live close to one of our physical office locations, our doors are open for you to come into the office as often as you'd like. 

Team:
The AI Serving Infrastructure team develops and maintains Reddit's high-performance model serving infrastructure, generating millions of model predictions per second with low latency, with a focus on GPU optimization. We collaborate closely with machine learning engineers to support a diverse range of models and partner with training and feature teams to deliver a seamless train-to-serve experience for efficient and reliable model deployment

As a Staff ML Infrastructure Engineer, AI Serving, you will be responsible for designing, building, and maintaining the infrastructure and platform that powers our search capabilities. You will collaborate with Backend Architects and Product ML teams to shape the roadmap for ML infrastructure and platforms, while providing ongoing support to scale the platform and meet evolving needs for state-of-the-art models.

How You'll Have Impact
You will build systems and tools that enable machine learning engineers (MLEs) and data scientists (DSs) and continuously improve the ML software development lifecycle, enabling the continuous iteration and improvement of systems that use ML techniques, including Deep Learning, Natural Language Processing, Recommendation Systems, Representation Learning, and Computer Vision.

• Lead the building, testing, and maintenance of ML infrastructure at Reddit
• Propose, design, and implement high-performance ML Infra solutions that significantly advance the deployment of models that serve millions of redditors a seamless experience for MLEs
• Design and implement solutions that significantly advance the ML Infrastructure
• Collaborate closely with XFN partner teams/internal customers.
• Work with management on team goal setting, planning, and de-risk project execution
• Mentor other team members in adopting a rigorous DevOps approach to maintain and/or improve ML infra components and services health and quality

Who You Might Be:

• 8+ years of work experience in a production software development environment
• Experience working on large-scale ML Systems
• Experience building production-quality code incorporating testing, evaluation, and monitoring using object-oriented programming, including experience in golang, Python and Scala.
• Experience with Kubernetes
• Knowledge of maintaining or developing applications using large-scale data stack, e.g, BigQuery, GraphQL, Kafka, Flink, Cassandra, Redis, Search Databases 
• Comfortable with distributed systems, big data (Petabyte scale) and data-intensive systems
• Strong organizational & communication skills

Benefits:

• Comprehensive Healthcare Benefits
• 401k Matching
• Workspace benefits for your home office
• Personal & Professional development funds
• Family Planning Support
• Flexible Vacation (please use them!) & Reddit Global Wellness Days
• 4+ months paid Parental Leave
• Paid Volunteer time off

#LI-DB1 #LI-Remote