Senior/Staff AI Research Engineer Location: London (Hybrid) Salary: £150-250k base + generous equityAre you excited to drive AI drug discovery forward by scaling SOTA models and optimizing neural nets on GPUs?Qualifications• 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• Experience with GPU architectures, cloud platforms, and hardware tradeoffs• Familiarity with low-level hardware tuning and custom CUDA kernel developmentKey Responsibilities• 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 advancementsBenefits• Generous sock options• Competitive base• Private healthcare• Generous pension contributions• Contribute to impactful research at the AI-healthcare frontier• Access to professional growth via conferences, workshops, and training• Inclusive, collaborative culture focused on innovation and continuous learningIf you meet 60-70% of the requirements I would love to hear from you. Click the Easy Apply button!