Research engineer- Consultant Assignment

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.

Our vibrant Gothenburg site brings together more than 2,900 employees from over 50 countries, creating a truly inspiring and collaborative environment. Here, history and scientific breakthroughs unite to drive new discoveries. We believe the diversity of our team is essential for turning bold ideas into reality.

At AstraZeneca, your engineering skills will directly influence our technology platforms and help shape the next generation of transformative medicines. Are you ready to make a difference? Join us and be part of our journey.

What You'll Do:
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/
• 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.

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