Principal Machine Learning Engineer, Infrastructure (LLM)

$188,200 - $349,400/Yr

Genentech - New York, NY

Full-time - Principal

New York, NY

Chemical Manufacturing

About the position

The Principal Machine Learning Engineer, Infrastructure (LLM) at Genentech will play a pivotal role in advancing drug discovery through the development of machine learning infrastructure. This position focuses on designing, constructing, and optimizing large-scale distributed systems, particularly for large language models (LLMs) and AI systems. The engineer will contribute to cutting-edge research, develop innovative algorithms, and collaborate with cross-functional teams to tackle complex challenges in the life sciences.

Responsibilities

• Contribute to cutting-edge research in machine learning, including algorithm and method development for drug discovery and large language models (LLMs).
• Develop and maintain robust tools for large-scale ML experiments, focusing on deployment, optimization, and scalability of compute-intensive LLM training.
• Design, implement, and scale dynamic, distributed, and high-performance machine learning systems, including models deployed as APIs and retrieval augmented generation (RAG) systems.
• Collaborate with cross-functional teams across Prescient Design, gCS, gRED, and global Roche to solve complex problems in life sciences.

Requirements

• MS or PhD in Computer Science, Statistics, Machine Learning, or related field, or equivalent experience.
• 6+ years of industry experience relating to distributed systems and machine learning.
• Proven expertise in designing and implementing large-scale machine learning models and infrastructure.
• Demonstrated success in technical leadership for teams of engineers.
• Practical hands-on experience with PyTorch in a production environment.
• Excellent programming skills.

Nice-to-haves

• Familiarity with large language models (LLMs) and their training is highly desirable.
• A keen interest in molecular design and a desire to contribute to scientific and computational advancements.

Benefits

• Relocation benefits are available for this opportunity.
• Discretionary annual bonus based on individual and Company performance.