Research Scientist, AI/ML Foundational Models

Takeda•Boston, MA

13d•Hybrid

About The Position

We are seeking Scientists to develop and deploy foundational AI models that will transform drug discovery across Takeda. As part of the AI/ML Foundation team, you will build large-scale models including large language models (LLMs), diffusion models, and multimodal architectures that integrate diverse data types—omics, biomedical imaging, protein 3D structures, and molecular representations. This role requires deep expertise in modern deep learning architectures combined with foundational knowledge of biology, chemistry, and disease biology to ensure models are scientifically grounded and impactful. You will train models from scratch, fine-tune pre-trained models for Takeda-specific applications, and deploy foundation model capabilities that accelerate discovery across all therapeutic platforms.

Requirements

PhD degree in a scientific discipline (or equivalent), or MS with 6+ years relevant experience, or BS with 8+ years relevant experience
Deep expertise in modern deep learning architectures including transformers, diffusion models, and/or generative models.
Strong experience training large-scale models with proficiency in PyTorch and distributed training frameworks.
Foundational knowledge of biology, chemistry, or disease biology sufficient to guide scientifically meaningful model development.
Experience with at least one of: protein language models (ESM, ProtTrans), molecular generative models, or biomedical vision models.
Experience with cloud computing (AWS, GCP) and GPU cluster training at scale.
Proficiency in Python and ML libraries (PyTorch, TensorFlow, scikit-learn); familiarity with Unix tools.
Excellent collaboration and communication skills.

Nice To Haves

Experience building or fine-tuning foundation models in pharmaceutical or life sciences settings.
Expertise in multimodal learning integrating text, images, and structured molecular data.
Experience with omics data analysis (genomics, transcriptomics, proteomics) and knowledge graph
Familiarity with protein structure prediction and 3D molecular representations.
Publications in top-tier ML venues (NeurIPS, ICML, ICLR) or computational biology journals.
Experience with model compression, efficient inference, or production deployment of large models.
Strong background in large-scale data integration and multimodal modeling for biological systems.
Ability to lead cross-functional initiatives and mentor junior scientists.
Experience in translating computational insights into experimental strategies.
Strong publication record or demonstrated thought leadership in AI for biology and molecular design.
Comfort working in fast-paced, innovation-driven environments with evolving priorities.

Responsibilities

Develop and train foundational AI models (LLMs, diffusion models, flow-matching architectures) for drug discovery applications, with capability to pre-train on large-scale scientific corpora and molecular datasets.
Fine-tune and adapt pre-trained foundation models (protein language models, chemical LLMs, vision transformers) for Takeda-specific applications in target identification, disease modeling, and molecular design and discovery.
Build multimodal foundation models integrating diverse data types including omics (genomics, transcriptomics, proteomics), biomedical imaging, protein 3D structures, and molecular representations.
Apply and extend state-of-the-art approaches including graph neural networks, transformer-based protein language models, and multimodal learning frameworks.
Apply domain expertise in biology, chemistry, and/or disease biology to guide model architecture decisions, training data curation, and evaluation strategies ensuring scientific validity.
Implement state-of-the-art generative architectures (diffusion, score-based models, autoregressive transformers) for molecular generation, protein design, and multi-objective optimization.
Collaborate with computational scientists across domains to deploy foundation models that address diverse discovery needs across small molecules, biologics, and emerging modalities.
Stay current with advances in foundation models, generative AI, and multimodal learning; contribute to internal knowledge sharing and external publications.

Benefits

U.S. based employees may be eligible to participate in medical, dental, vision insurance, a 401(k) plan and company match, short-term and long-term disability coverage, basic life insurance, a tuition reimbursement program, paid volunteer time off, company holidays, and well-being benefits, among others.
U.S. based employees are also eligible to receive, per calendar year, up to 80 hours of sick time, and new hires are eligible to accrue up to 120 hours of paid vacation.

Stand Out From the Crowd

Upload your resume and get instant feedback on how well it matches this job.

Upload and Match Resume