Senior Machine Learning Engineer

Summary

Join Reddit's team as an exceptional AI Engineer specializing in LLM training and optimization. You will design, develop, and optimize state-of-the-art language models for various applications, collaborating with cross-functional teams. This pivotal role requires expertise in deep learning frameworks, distributed training, and natural language processing. You will contribute to responsible AI development and stay updated on the latest research advancements. Reddit offers a flexible work environment with options for remote work and in-office collaboration. The company is committed to building a diverse and inclusive workforce.

Requirements

• 5+ years of relevant MLE experience in natural language processing, deep learning, and AI model development
• Strong background in Python programming and 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

• Design, develop, and optimize large language models for various natural language processing tasks
• Implement and maintain training pipelines, leveraging distributed training and optimizing for performance and efficiency
• Collaborate with cross-functional teams to gather requirements, define model architectures and iterate on model development
• Conduct model evaluations, performance analysis, and optimization to improve model accuracy and reduce biases
• Stay up-to-date with the latest research and advancements in the field of natural language processing, multimodal signals, and large language models
• Contribute to the development of best practices, guidelines, and ethical AI principles for responsible LLM development and deployment

Benefits

• Pension Savings plan
• Medical Plan
• Short term sickness benefits
• WIA excess and WGA gap insurance
• Workspace benefits for your home office
• Personal & Professional development funds
• Family Planning Support
• Flexible Vacation & Reddit Global Days Off