Location: Montreal
Job Number: J Job Type: Full Time Job Category:
Information Technology Salary:
Open
Sr Machine Learning Engineer – LLM/RL - Chemistry and Drug Discovery
This is a leading company in artificial intelligence for drug discovery that has launched multiple software products, including a target identification platform and an automated drug discovery platform. We are looking for a ML Engineer to join the team. You will work with product teams as well as top AI scientists in this field. The company focuses on AI-assisted systems that can identify novel drug targets for untreated diseases, assist in the development of new treatments, and eventually predict how well those treatments may perform in clinical trials.
It was ranked Top 5 AI companies in its potential for social impact according to NVIDIA and was named one of the global top 100 AI companies by CB Insights. They are looking to add someone to their team in their Abu Dhabi or Shanghai office. Full relocation support is available.
More about the role:
The company seeks a Senior Machine Learning Engineer (LLM/RL) to build and scale GenAI models for molecular tasks. You would be working on a comprehensive small-molecule drug discovery platform.
The hiring manager emphasized that this person should have strong experience in Reinforcement Learning and be skilled in training models on a multi-GPU environment. Knowledge of RL libraries is preferred.
Experience with diffusion models is also relevant but less prioritized.
Place of work
• [Office Address - Abu Dhabi]
• [Office Address - Shanghai]
• Reports to ML team leader
• Details about relocation are provided below.
Responsibilities
• Design and scale reinforcement learning fine-tuning workflows for LLMs, including RLVR/RLHF approaches.
• Build and optimize high-throughput transformer training pipelines across multi-GPU and multi-node environments.
• Adapt models to domain-specific drug discovery and molecular design tasks, such as molecular generation and property prediction.
• Lead experimental design, monitor core metrics, and communicate insights and findings through presentations and/or publications.
Education
, , or equivalent relevant experience in Computer Science, Machine Learning, Computational Chemistry, or a related quantitative discipline.
Required Skills & Experience
• ML Engineering:
At least five years of experience in ML engineering
• Transformers:
Expertise in training and deploying large transformer models using Py Torch
• RL Algorithms:
Hands-on experience with PPO/DPO/RLOO/GRPO and other related policy gradient and preference-based optimization methods
• Scalable Training Pipelines:
Proficiency in data, model, context, and pipeline parallelism
• Reward Integration:
Experience developing diverse reward functions and models
• Tooling &
Collaboration:
Excellent Python skills, experience with Hugging Face Transformers, TRL/RLlib/Verl, Deep Speed, vLLM, Docker, Sagemaker, and CI/CD workflows
Desirable Skills
• Senior Level: At least two years in a senior position
• Drug Discovery & Molecular Design:
Background in chemistry, cheminformatics, structural biology, or experience with molecular generative models
• Generative Models:
Knowledge of diffusion, flow-based, and other generative models
• Research Leadership:
Track record of machine learning publications
Relocation details for Abu Dhabi
• 0% income tax in UAE
• 1 month remote work anywhere in the world per year
• 6 months of company-covered housing, followed by 2 years of housing allowance (covers most costs)
• Housing suitable for a family (spouse + 2 children)
• Schooling support: 100% for 1 child, 75% for another (for 2 years)
• Covered airfare for relocation and travel
• Relocation package for moving costs negotiable
• Salary around $100K USD, tax-free; flexible for highly experienced candidates
Votre partenaire en recrutement – Your recruitment partner
Are you interested in this job?
