AI Scientist – Molecular Machine Learning

Marlabs; San Diego (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

Marlabs, a global AI and Digital Solutions Consulting firm, seeks an AI Scientist – Molecular Machine Learning to design and develop advanced AI/ML models for molecular property prediction, generation, and drug discovery. You will apply deep learning, graph neural networks, geometric learning, transformers, and generative AI to complex molecular datasets with chemists to drive discovery.

The candidate is an AI/ML-first scientist with strong model development skills and experience applying ML to

Qualifications

• Ph.D. in Computer Science, AI/ML, Computational Chemistry, Cheminformatics, Data Science, Physics, or related field with relevant experience applying AI/ML to molecular problems.
• Strong hands-on experience with PyTorch and/or JAX, including developing, training, and evaluating deep learning models.
• Demonstrated experience with Graph Neural Networks, Graph Transformers, geometric deep learning, and generative AI for molecular design or property prediction.
• Strong Python programming and cheminformatics experience, including RDKit or comparable toolkits.
• Experience applying AI/ML to molecular property prediction, molecular generation, drug discovery, or related molecular science applications.
• Experience in Linux and HPC environments; exposure to Schrödinger, MOE, GROMACS, OpenMM is valuable.

Responsibilities

• Design, develop, train, and evaluate deep learning models for molecular property prediction, generation, and drug discovery.
• Build molecular AI solutions using GNNs, Graph Transformers, geometric deep learning, and generative modeling.
• Develop scalable AI/ML workflows using Python, PyTorch and/or JAX, optimizing for accuracy and efficiency.
• Manage molecular datasets and cheminformatics workflows using RDKit, including data prep and feature generation.
• Develop generative modeling approaches for de novo molecular design and lead optimization.
• Collaborate with chemists to translate problems into AI/ML approaches and derive insights for discovery programs.
• Evaluate and improve models with rigorous experimentation and reproducible research practices.

Skills

PyTorch

JAX

Graph Neural Networks

Graph Transformers

Deep learning

Python

RDKit

Cheminformatics

Linux HPC

Education

Ph.D. in Computer Science

Tools

RDKit

Schrödinger

MOE

GROMACS

OpenMM

Job description

Marlabs, a global AI and Digital Solutions Consulting firm, delivers intelligent solutions across AI, data, analytics, and product engineering. Since 2000, we have partnered with some of the largest healthcare, life sciences, financial services, and government organizations worldwide. As we continue to expand our global footprint, we have an exciting opportunity for a highly skilled AI Scientist – Molecular Machine Learning to join our innovative and dynamic team.

AI Scientist – Molecular Machine Learning | About You

As an AI Scientist – Molecular Machine Learning, you will design and develop advanced AI/ML models that accelerate molecular property prediction, molecular generation, and drug discovery. You will apply deep learning, graph neural networks, geometric learning, transformers, and generative AI to complex molecular and chemical datasets, partnering closely with computational chemists and drug discovery scientists to solve high-impact scientific challenges. The ideal candidate is an AI/ML-first scientist with strong model development expertise and demonstrated experience applying machine learning to chemistry, molecular design, or drug discovery.

AI Scientist – Molecular Machine Learning | Day-to-Day

• Design, develop, train, and evaluate deep learning models for molecular property prediction, molecular generation, and drug discovery applications.
• Build molecular AI solutions using Graph Neural Networks (GNNs), Graph Transformers, geometric deep learning, and generative modeling techniques.
• Develop scalable AI/ML workflows using Python, PyTorch and/or JAX, optimizing models for accuracy, efficiency, and scientific relevance.
• Build and manage molecular datasets and cheminformatics workflows using RDKit and related tools, including molecular representation, preprocessing, feature generation, and model-ready dataset development.
• Develop generative modeling approaches for de novo molecular design, lead optimization, and molecular exploration, leveraging techniques such as diffusion models, flow matching, reinforcement learning, or related architectures.
• Collaborate with computational chemists, medicinal chemists, and drug discovery scientists to translate scientific problems into AI/ML approaches and generate actionable insights for molecular discovery programs.
• Evaluate and continuously improve molecular ML models using rigorous experimentation, validation, benchmarking, and reproducible research practices.

AI Scientist – Molecular Machine Learning | Skills & Experience

• Ph.D. in Computer Science, AI/ML, Computational Chemistry, Cheminformatics, Data Science, Physics, or a related technical or scientific discipline, with relevant experience applying AI/ML to molecular or scientific problems.
• Strong hands-on expertise with PyTorch and/or JAX, including developing, training, evaluating, and optimizing deep learning models.
• Demonstrated experience with Graph Neural Networks, Graph Transformers, geometric deep learning, and generative AI, particularly for molecular representation, property prediction, or molecular design.
• Strong Python programming and cheminformatics experience, including RDKit or comparable toolkits, with experience working with complex molecular datasets.
• Experience applying AI/ML to molecular property prediction, molecular generation, drug discovery, ADMET modeling, lead optimization, or related molecular science applications.
• Experience working within Linux and high-performance computing environments, with exposure to tools such as Schrödinger, MOE, GROMACS, OpenMM, or other molecular modeling platforms considered valuable.