Staff Software Engineer, ML Understanding

Summary

Join Reddit's ML Understanding team as a Software Engineer and lead the design, development, and maintenance of backend and data infrastructure for AI/ML systems. You will architect scalable solutions, collaborate with cross-functional teams, and mentor junior engineers. This role requires 7+ years of experience in NLP and AI model integration, strong Python skills, and expertise in distributed training frameworks. Reddit offers a flexible work environment with options for remote work and in-office collaboration, along with comprehensive benefits including a group pension scheme, private medical and dental coverage, and various other support programs.

Requirements

• 7+ years of relevant experience as a software engineer involved in projects related with natural language processing and AI models integration in products
• Strong background in Python programming and some experience with deep learning frameworks like TensorFlow, PyTorch, or Hugging Face Transformers
• Expertise in distributed training frameworks (e.g., Ray Training, PyTorch Distributed), and efficient utilization of hardware resources
• Proficiency in data preprocessing, tokenization, embeddings, and language modeling techniques
• Passion for developing scalable, well-designed, and responsible AI solutions that positively impact society
• Excellent communication and collaboration skills, with the ability to discuss complex technical topics with diverse teams
• Entrepreneurial spirit, self-motivation, and a bias towards action in fast-paced environments

Responsibilities

• Lead the design, development, and maintenance of the backend and data infrastructure for efficient, accurate, and optimized deployment of AI / ML based systems
• Architect and implement scalable AI / ML based solutions that can handle both the long term growth in the size of our data, and the scalability of the support in accessing it across a growing number of needs within the company
• Collaborate with other teams and functions to work towards consolidating all insights about Reddit’s data to be powered by the same platform
• Oversee data storage, retrieval, indexing, and query optimization strategies to ensure fast and reliable access to our data. This may include migrating the data to the right datastore given the company’s growing needs
• Collaborate with cross-functional teams to gather requirements, define system architectures and support system development
• Provide technical guidance and mentorship to engineers on the team, fostering a culture of learning and innovation
• Proactively identify technical challenges and propose innovative solutions that align with the team's goals

Benefits

• Group Personal Pension Scheme with Employer match
• Private Medical and Dental Scheme
• Income Replacement Programs
• Family Planning Support
• Gender-Affirming Care
• Mental Health & Coaching Benefits
• Bike to Work scheme
• Flexible Vacation & Reddit Global Days Off