Staff Software Engineer, AI Serving Infrastructure

Reddit; United States; Remote; USD 230,000 - 322,000; Full time

Job summary

A leading company seeks a Staff ML Infrastructure Engineer to advance Reddit's model serving abilities. This role focuses on designing and optimizing infrastructure for machine learning, working with teams across product and architecture. Comprehensive benefits and a competitive salary are offered, fostering a supportive and flexible working environment.

Benefits

Comprehensive Healthcare Benefits

401k Matching

Workspace benefits for home office

Family Planning Support

Flexible Vacation

Paid Volunteer time off

Qualifications

• Leading ML infrastructure development and maintenance.
• Experience with large-scale ML systems essential.
• Proficient in building production-quality code.

Responsibilities

• Design and maintain infrastructure for ML capabilities.
• Collaborate on roadmap for ML infrastructure.
• Mentor team members on DevOps for ML services.

Skills

Object-oriented programming

Distributed systems

Big data

Kubernetes

ML Systems

Communication

Organizational skills

Education

8+ years of work experience in software development

Tools

Golang

Python

Scala

BigQuery

Kafka

Flink

Cassandra

Redis

Job description

Reddit is a community of communities. It’s built on shared interests, passion, and trust and is home to the most open and authentic conversations on the internet. Every day, Reddit users submit, vote, and comment on the topics they care most about. With 100,000+ active communities and approximately 101M+ daily active unique visitors, Reddit is one of the internet’s largest sources of information. For more information, visit redditinc.com.

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

Pay Transparency:

This job posting may span more than one career level.

In addition to base salary, this job is eligible to receive equity in the form of restricted stock units, and depending on the position offered, it may also be eligible to receive a commission. Additionally, Reddit offers a wide range of benefits to U.S.-based employees, including medical, dental, and vision insurance, 401(k) program with employer match, generous time off for vacation, and parental leave. To learn more, please visit https://www.redditinc.com/careers/.

To provide greater transparency to candidates, we share base pay ranges for all US-based job postings regardless of state. We set standard base pay ranges for all roles based on function, level, and country location, benchmarked against similar stage growth companies. Final offer amounts are determined by multiple factors including, skills, depth of work experience and relevant licenses/credentials, and may vary from the amounts listed below.

The base pay range for this position is: $230,000 — $322,000 USD

Reddit is proud to be an equal opportunity employer, and is committed to building a workforce representative of the diverse communities we serve. Reddit is committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures. If, due to a disability, you need an accommodation during the interview process, please let your recruiter know.