AI Scientist – Molecular Machine Learning

Marlabs; United States; Remote; USD 140,000 - 190,000; Full time

Job summary

Marlabs is hiring 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, GNNs, transformers, and generative AI to complex molecular data, collaborating with computational and medicinal chemists on high-impact projects.

The ideal candidate has a PhD and strong model development expertise in chemistry-related AI, with experience in RDKit and Python.

Qualifications

• PhD in a relevant technical or scientific discipline with molecular/AI experience.
• Hands-on PyTorch and/or JAX for model development and optimization.
• Expertise in GNNs, Graph Transformers, geometric deep learning, and generative AI for molecules.
• Strong Python and cheminformatics skills, with RDKit experience.
• Experience applying AI/ML to molecular property prediction, generation, or drug discovery.

Responsibilities

• Design, train, and evaluate deep learning models for molecular property prediction and drug discovery.
• Build molecular AI solutions using GNNs, Graph Transformers, and diffusion/generative techniques.
• Develop scalable AI workflows in Python with PyTorch/JAX, optimize for accuracy and efficiency.
• Assemble molecular datasets and cheminformatics workflows using RDKit.
• Collaborate with chemists to translate problems into AI/ML approaches.
• Evaluate models with rigorous experimentation and reproducible research practices.

Skills

PyTorch / JAX

Graph Neural Networks

Geometric deep learning

Python programming

RDKit

Molecular ML applicability

Education

Ph.D. in Computer Science / AI / Computational Chemistry / Cheminformatics / Data 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.

Employment Type: Full-time

Location: Remote - USA

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.