Arbetsbeskrivning

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 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 AstraZeneca is one of the world's most exciting bio-pharmaceutical companies. From scientists to sales, lab techs to legal, we're on a mission to turn ideas into life-changing medicines that improve patients' lives and benefit society. We need great people who share our passion for science and have the drive and determination to meet the unmet needs of patients around the world. About Hays: Hays is a global recruitment company with extensive experience in matching the right skills with the right assignments. As a consultant with us, you become part of our network and gain access to exciting assignments at leading companies.