Senior Machine Learning Engineer

Senior Machine Learning Engineer

Cambridge, UK

Competitive Salary and benefits on offer.

Join AstraZeneca's AI Strategy & Innovation team, a key player in the delivery of the ORD AI Strategy in 2024. This multi-million-dollar investment aims to revolutionize the R&D value chain through AI, bringing deeper insights, automation, and scale to our portfolio. As a Senior Machine Learning Engineer, you will have the opportunity to shape AstraZeneca's digital journey and make a significant contribution to our growth.

Accountabilities:

In this role, you will focus on developing deep learning models for AstraZeneca R&D to expedite clinical trials. You will design models to segment tumours from CT scans, predict patient responses to medicine, and forecast toxicity of different compounds. Your work will involve conducting research on machine learning models, applying them to clinical settings, and optimizing them for production. The models you develop will directly impact patients’ lives and must be designed with extreme care, rigorous development practices, robust testing, and performance auditing.

Essential Skills/Experience:

• Develop interactive and fully automatic models that segment tumours from CT, PET and MRI scans for different tumour types.
• Align different images together.
• Develop multi-modal prediction models for tumour segmentation and patient response prediction.
• Design and implement large model training pipelines that can train models using many GPUs.
• Design large language models that use genetic data.
• Design models to predict toxicity of compounds from tabular and multi-dimensional image.
• Develop segmentation pipelines to extract individual cells from multi-dimensional image.
• Develop transformer-based architectures to handle unordered sequence of data.
• Hands-on experience in training, debugging and deploying deep learning models.
• Good coding skill in machine learning products, not only research prototypes.
• Ability to read a lot of research papers in deep learning, digest research ideas and improve them into products.
• Theoretical understanding of deep computer vision models.
• Good experience with Python, PyTorch and PyTorch Lightning.

Desirable skills/experience

• PhD in machine learning, computer science, math, physics.
• Established track record of publications in top tier conferences or journals.
• Experience in mentoring junior members.
• Experience working with medical data, such as CT scans, H&E slices, genetic data.
• Experience in model inference speed optimization and model distillation.
• Experience in working with Contrastive learning approaches such as DINO, SimCLR and BYOL.

When we put unexpected teams in the same room, we unleash bold thinking with the power to inspire life-changing medicines. In-person working gives us the platform we need to connect, work at pace and challenge perceptions. That's why we work, on average, a minimum of three days per week from the office. But that doesn't mean we're not flexible. We balance the expectation of being in the office while respecting individual flexibility. Join us in our unique and ambitious world.

At AstraZeneca, we are committed to leading the transformation by fusing our digital and data capabilities with backing from the business. We apply our skills for good, positively impacting patients across the world. We are a diverse team of technical experts who never stand still. We challenge, innovate and break away from the norm to find bold new ways of approaching the everyday. We understand how to add value - to ourselves, the business and our patients. We partner across the business, leveraging our leading technologies and exploring data to make improved decisions. We're not only backed by huge investment, we also feel valued. And as our data and digital capabilities expand every day, we know we're playing our part in accelerating AstraZeneca's growth.

Ready to make a difference? Apply now and join us in our mission to create a better future for patients worldwide!

Open date: 11/10/2024

Closing date: 25/10/2024

Competitive Salary and Benefits on offer.

AstraZeneca embraces diversity and equality of opportunity. We are committed to building an inclusive and diverse team representing all backgrounds, with as wide a range of perspectives as possible, and harnessing industry-leading skills. We believe that the more inclusive we are, the better our work will be. We welcome and consider applications to join our team from all qualified candidates, regardless of their characteristics. We comply with all applicable laws and regulations on non-discrimination in employment (and recruitment), as well as work authorisation and employment eligibility verification requirements.