Staff Research Engineer

Senior/Staff AI Research Engineer Location: London (Hybrid) Are you excited to drive AI drug discovery forward by scaling SOTA models and optimizing neural nets on GPUs? Bachelor\’s/ Master\’s or PhD in Computer Science, Engineering, or a related field • 5+ years of experience deploying AI/ML models in production settings • Proficient in Python, C++, and deep learning frameworks like PyTorch or Jax • Skilled in distributed training (e.g., DDP, FSDP) and model performance optimization • Develop cutting-edge AI models for drug discovery using deep learning frameworks • Scale and optimize distributed training of large AI models on GPUs • Enhance performance of AI models during both training and inference stages • Communicate findings through documentation, presentations, and stay updated on AI advancements Private healthcare • Generous pension contributions • Contribute to impactful research at the AI-healthcare frontier • Access to professional growth via conferences, workshops, and training •