Senior Ai Engineer

AstraZeneca

Beijing, China
5+ years distributed deep learning experience
Strong pytorch production-grade expertise
Multi-node multi-gpu training templates
The Senior AI Engineer will design and deliver the distributed training infrastructure to enable Discovery AI scientists to train foundation models

Job Summary

  • The Senior AI Engineer will design and deliver the distributed training infrastructure to enable Discovery AI scientists to train foundation models.
  • This role serves as the bridge between IT's hardware infrastructure and Discovery's scientific workloads within AstraZeneca's Beijing AI Center.
  • Candidates must establish AI engineering standards including naming conventions, experiment configuration, and reproducibility criteria for the team.

Matching Summary

The Senior AI Engineer will design and deliver the distributed training infrastructure to enable Discovery AI scientists to train foundation models.

Skills & Requirements

Must-have

  • 5+ years distributed deep learning experience
  • Strong PyTorch production-grade expertise
  • Multi-node multi-GPU training templates
  • Kubernetes job scheduling experience
  • Transformer model optimization skills

Nice-to-have

  • Parameter-efficient fine-tuning methods
  • Reinforcement learning training infrastructure
  • AWS China or Alibaba Cloud experience
  • NVIDIA H20 or H100-series GPU familiarity
  • Biopharma domain knowledge in drug discovery

Key Requirements

  • 5+ years experience in distributed deep learning
  • Hands-on experience with DDP, FSDP, or DeepSpeed
  • Experience setting AI/ML engineering standards for teams
  • Ability to work full-time in Beijing

Work Rights

Must be able to work full-time in Beijing

Tailored Resume

Cover Letter