Associate Principal AI Engineer

Join the Center for Artificial Intelligence (CAI) at AstraZeneca, a lab dedicated to applying machine learning research to the toughest challenges in the medical field. We are a diverse team of experts from various fields including Machine Intelligence, Data Science, Real World Evidence and Bioinformatics.



As an Associate Principal AI Engineer, you will help deliver innovative AI research as elegant, enterprise-ready products. You will be responsible for developing efficient and scalable processes for the distributed training of large machine learning (ML) models, as well as automating our processes to accelerate project delivery and reduce costs and environmental impact. The role is to take the most important research done at the CAI and deliver it as elegantly engineered fully compliant, enterprise ready products. This role is based at our vibrant R&D site in Gothenburg, Sweden.



Key Accountabilities

*

Take technical leadership of projects regarding software design, reproducibility and efficiency.

*

Ownership of software engineering standards within the team, the documentation of these standards along with the development of artifacts that make it simple to apply to new or existing projects.

*

Developing efficient and scalable processes for the distributed training of large machine learning models in both cloud and on-prem infrastructure. This will involve working with both internal infrastructure teams and external vendors and research institutes.

*

Accelerating project delivery, driving down cost and environmental impacts of the CAI by DevOps, GitOps and MLOps automation, thereby reducing the need for human intervention in our processes.



Essential Skills and Experience

*

A master's degree or PhD in a relevant field.

*

A track-record of implementing software engineering best practices for multiple use cases.

*

Professional experience automating the entire machine learning model lifecycle.

*

Advanced proficiency in Python and common scientific libraries (e.g. Numpy, Pandas).

*

Experience using PyTorch to implement machine learning models and training.

*

Demonstratable experience with optimization of distributed training of machine learning models.

*

Experience with GitHub for source control and GitHub Actions for CI/CD.

*

Experience using Kubernetes to implement machine learning pipelines and the Kubeflow ecosystem.

*

Experience with deployment of solutions on a cloud vendor such as AWS, GCP or Azure.

*

Experience partnering and/or collaborating with teams that have different competences.



Desirable Skills and Experience

*

Relevant certifications.

*

Experience working in a domain subject to regulatory oversight.

*

Experience working in scientific research environment.

*

Understanding of the pharmaceutical industry and its processes.





Why AstraZeneca

At AstraZeneca, we follow the science to explore and innovate. We are working towards treating, preventing, modifying and even curing some of the world's most complex diseases. Here we have the potential to grow our pipeline and positively impact the lives of billions of patients around the world. We are committed to making a difference. We have built our business around our passion for science. Now we are fusing data and technology with the latest scientific innovations to achieve the next wave of breakthroughs.





So, what's next?

Are you ready to be part of a team where you are empowered to follow the science? Ready to make an impact? Apply now and join us in our mission to improve patient access to healthcare globally!



We welcome your application no later than June 19th, 2024. Applications will be reviewed continuously and interviews may be held on a continuous basis.