Machine Learning Engineer / Scientist

Are you a machine learning engineer or scientist excited by building and scaling modern AI systems for real-world impact? Do you want to enable the training and inference of large-scale deep learning models for drug discovery? We are seeking a 15-month consultant to join the innovative MolecularAI department at AstraZeneca in Gothenburg, Sweden.

Responsibilities

You will drive impact across multiple projects with a focus on AI engineering and platform development. Your responsibilities will include:

• Optimize the training and inference of large-scale deep learning models for drug design on state-of-the-art HPC platforms
• Design, conduct and evaluate scientific experiments together with our multi-disciplinary research team to help us build the next generation of foundational chemical models.
• Extend and maintain scientific codebases written in Python that forms the backbone of our AI platforms. See https://github.com/molecularai/ (https://github.com/molecularai/)
• Collaborate cross-functionally with researchers in data science, software engineers and drug discovery domain experts

This is an on-site role based in Gothenburg, Sweden where you expect to present on site three days per week.

Your Profile

Essential Qualifications

We believe you are passionate about building engineering solutions for life sciences. You have:

• Degree in computer science, engineering, chemistry, physics, mathematics or related discipline, or equivalent industry experience
• Hands-on experience training models with PyTorch or equivalent frameworks
• Excellent programming skills, preferably in Python. Track record of developing sizable software solutions and familiarity with scientific packages (Numpy, Pandas, etc.)
• Proven ability to work inclusively within multidisciplinary teams, with a focus on building a positive and collaborative team culture.

Desirable Requirements

You would benefit from having experience or familiarity with:

• Experience with high-performance computational platforms and schedulers such as SLURM
• Experience writing ETLs for large data sets
• Background in life sciences or physical sciences
• Research and outreach contribution to open-source projects, publications, or participation in online communities

About the Organisation

This is a consultant assignment at AstraZeneca in Gothenburg for 15 months, starting in April 2026. During the time you will be hired by QRIOS.

Please apply by registering your CV here on our website. Click on the "Apply" button to the right. It is our recommendation that you register as much information as possible and that you attach a complete CV document. Please check that your correct email address is registered.