AI Research engineer

AI Research engineer - 15 months consultant assignment - AstraZeneca 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 r