Machine Learning Scientist, Scientific Reasoning Models, AI for Drug Discovery

Roche's AI for Drug Discovery (AIDD) group is seeking a Machine Learning Scientist to join the Foundation Models team within Prescient Design (gRED). The role involves contributing to internal reasoning Large Language Models (LLMs) and enabling them to succeed at relevant drug discovery tasks, including biomolecular design. The Machine Learning Scientist will design, implement, and improve large-scale distributed machine learning systems, develop strategies to improve model performance, and collaborate with researchers to translate ideas into scalable, production-ready systems.

Requirements

• BS/MS in Computer Science, Statistics, Mathematics, Physics, or a related quantitative field
• 2+ years of relevant work experience or Ph.D. with 0-2 years of relevant work experience
• Experience developing and training large-scale machine learning models
• Strong software engineering skills and experience working with high-performance computing systems
• Publication record at top-tier venues (e.g., NeurIPS, ICLR, ICML)

Benefits

• #ComputationCoE
• #tech4lifeComputationalScience
• #tech4lifeAI
• Equal opportunity employer
• Discretionary annual bonus may be available