Staff Machine Learning Engineer, ML Understanding

Summary

Join Reddit's ML Understanding Team as a Staff Machine Learning Engineer and develop cutting-edge content-based embeddings to represent content, communities, and users. Design and deploy content understanding pipelines leveraging state-of-the-art ML models and LLMs. Conduct model evaluations and optimizations to improve accuracy and reduce bias. Collaborate with cross-functional teams to align product and technical roadmaps. Stay at the forefront of AI research, introducing new modeling paradigms. Contribute to best practices and ethical AI principles. Mentor and guide other ML engineers. Drive technical discussions and contribute to long-term ML planning. This fully remote-friendly role offers significant impact on Reddit's user experience and product development.

Requirements

• 7+ years of relevant MLE experience in AI model development and deep learning. Experience in natural language processing, image, and video understanding is a plus
• Expert at developing and productionizing models using TensorFlow, PyTorch, or Hugging Face Transformers
• Experience building production-quality code incorporating testing, evaluation, and monitoring using object-oriented programming, including experience in Python, Golang
• Experience with distributed training frameworks (e.g., Ray Training, PyTorch Distributed), and efficient utilization of hardware resources
• 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

Responsibilities

• Develop state-of-the-art content-based embeddings to represent content, communities, and users
• Design and deploy content understanding pipelines that leverage state-of-the-art ML models, small language models, and the newest LLMs
• Conduct model evaluations, performance analysis, and optimization to improve accuracy and reduce bias
• Collaborate with cross-functional teams to align product and technical roadmaps and unlock key future ML capabilities
• Stay at the forefront of AI research, evaluating and introducing new modeling paradigms to keep Reddit’s ML ecosystem at the cutting edge
• Contribute to the development of best practices, guidelines, and ethical AI principles for responsible LLM development and deployment
• Mentor and guide senior and mid-level ML engineers, fostering a culture of excellence, innovation, and knowledge sharing
• Drive technical discussions, present findings to leadership, and contribute to long-term ML planning and decision-making

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